Spaces:
Running
Running
Update infer_zipvoice.py
Browse files- infer_zipvoice.py +3 -1
infer_zipvoice.py
CHANGED
|
@@ -242,7 +242,9 @@ model_defaults = {
|
|
| 242 |
},
|
| 243 |
}
|
| 244 |
|
| 245 |
-
device = torch.device("cuda", 0)
|
|
|
|
|
|
|
| 246 |
|
| 247 |
print("Loading model...")
|
| 248 |
model_config = "config.json"
|
|
|
|
| 242 |
},
|
| 243 |
}
|
| 244 |
|
| 245 |
+
# device = torch.device("cuda", 0)
|
| 246 |
+
|
| 247 |
+
device = torch.device("cpu")
|
| 248 |
|
| 249 |
print("Loading model...")
|
| 250 |
model_config = "config.json"
|