tts/layers/xtts/stream_generator.py: remove duplicate import + code

coqui-ai · Oct 23, 2023 · 419dc71 · 419dc71
1 parent 77f0655
commit 419dc71
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/TTS/tts/layers/xtts/stream_generator.py b/TTS/tts/layers/xtts/stream_generator.py
@@ -1008,10 +1008,8 @@ def init_stream_support():
 
 
 if __name__ == "__main__":
-    from transformers import AutoModelForCausalLM, AutoTokenizer, PreTrainedModel
-
-    PreTrainedModel.generate = NewGenerationMixin.generate
-    PreTrainedModel.sample_stream = NewGenerationMixin.sample_stream
+    from transformers import AutoModelForCausalLM, AutoTokenizer
+    init_stream_support()
     model = AutoModelForCausalLM.from_pretrained(
         "bigscience/bloom-560m", torch_dtype=torch.float16
     )