Zhyw commited on
Commit
dd60d9d
·
verified ·
1 Parent(s): af9f94b

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=True)
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):