Spaces:
Paused
Paused
Update src/main.py
Browse files- src/main.py +1 -1
src/main.py
CHANGED
|
@@ -247,7 +247,7 @@ def get_audio_file(song_input, is_webui, input_type, progress):
|
|
| 247 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 248 |
compute_half = True if torch.cuda.is_available() else False
|
| 249 |
config = Config(device, compute_half)
|
| 250 |
-
hubert_model = load_hubert("cuda", config.is_half,
|
| 251 |
print(device, "half>>", config.is_half)
|
| 252 |
|
| 253 |
# @spaces.GPU(enable_queue=True)
|
|
|
|
| 247 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 248 |
compute_half = True if torch.cuda.is_available() else False
|
| 249 |
config = Config(device, compute_half)
|
| 250 |
+
hubert_model = load_hubert("cuda", config.is_half, None)
|
| 251 |
print(device, "half>>", config.is_half)
|
| 252 |
|
| 253 |
# @spaces.GPU(enable_queue=True)
|