Spaces:
Running on Zero
Running on Zero
Update mossttsrealtime/streaming_mossttsrealtime.py
Browse files
mossttsrealtime/streaming_mossttsrealtime.py
CHANGED
|
@@ -102,7 +102,7 @@ class MossTTSRealtimeInference:
|
|
| 102 |
if not self._should_compile_local_transformer:
|
| 103 |
return self._generate_local_transformer_impl
|
| 104 |
if self._compiled_local_transformer is None:
|
| 105 |
-
self._compiled_local_transformer = torch.compile(self._generate_local_transformer_impl, fullgraph=
|
| 106 |
return self._compiled_local_transformer
|
| 107 |
|
| 108 |
def reset_generation_state(self, keep_cache: bool = True):
|
|
|
|
| 102 |
if not self._should_compile_local_transformer:
|
| 103 |
return self._generate_local_transformer_impl
|
| 104 |
if self._compiled_local_transformer is None:
|
| 105 |
+
self._compiled_local_transformer = torch.compile(self._generate_local_transformer_impl, fullgraph=False)
|
| 106 |
return self._compiled_local_transformer
|
| 107 |
|
| 108 |
def reset_generation_state(self, keep_cache: bool = True):
|