فهرست منبع

feat:修改mode

zhaohaipeng 1 ماه پیش
والد
کامیت
9a43060d43
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      fish_speech/models/text2semantic/inference.py

+ 1 - 1
fish_speech/models/text2semantic/inference.py

@@ -523,7 +523,7 @@ def init_model(checkpoint_path, device, precision, compile=False):
             decode_one_token,
             backend="inductor" if torch.cuda.is_available() else "aot_eager",
             mode="default" if torch.cuda.is_available() else None,
-            fullgraph=False,
+            fullgraph=True,
         )
 
     return model.eval(), decode_one_token