Spaces:
Running
on
Zero
Running
on
Zero
Update cosyvoice/cli/model.py
Browse files- cosyvoice/cli/model.py +1 -1
cosyvoice/cli/model.py
CHANGED
|
@@ -379,7 +379,7 @@ class CosyVoice2Model(CosyVoiceModel):
|
|
| 379 |
self.hift_cache_dict.pop(this_uuid)
|
| 380 |
if torch.cuda.is_available():
|
| 381 |
torch.cuda.empty_cache()
|
| 382 |
-
torch.cuda.current_stream().synchronize()
|
| 383 |
|
| 384 |
|
| 385 |
class CosyVoice3Model(CosyVoice2Model):
|
|
|
|
| 379 |
self.hift_cache_dict.pop(this_uuid)
|
| 380 |
if torch.cuda.is_available():
|
| 381 |
torch.cuda.empty_cache()
|
| 382 |
+
# torch.cuda.current_stream().synchronize()
|
| 383 |
|
| 384 |
|
| 385 |
class CosyVoice3Model(CosyVoice2Model):
|