Spaces:
Runtime error
Runtime error
DeathDaDev
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ import yaml
|
|
18 |
|
19 |
CUDA_AVAILABLE = torch.cuda.is_available()
|
20 |
|
21 |
-
snapshot = snapshot_download(repo_id='hexgrad/kokoro', allow_patterns=['*.pt', '*.pth', '*.yml']
|
22 |
config = yaml.safe_load(open(os.path.join(snapshot, 'config.yml')))
|
23 |
|
24 |
models = {device: build_model(config['model_params'], device) for device in ['cpu'] + (['cuda'] if CUDA_AVAILABLE else [])}
|
|
|
18 |
|
19 |
CUDA_AVAILABLE = torch.cuda.is_available()
|
20 |
|
21 |
+
snapshot = snapshot_download(repo_id='hexgrad/kokoro', allow_patterns=['*.pt', '*.pth', '*.yml'])
|
22 |
config = yaml.safe_load(open(os.path.join(snapshot, 'config.yml')))
|
23 |
|
24 |
models = {device: build_model(config['model_params'], device) for device in ['cpu'] + (['cuda'] if CUDA_AVAILABLE else [])}
|