瀏覽代碼

Fix inference error

Lengyue 2 年之前
父節點
當前提交
45b8ed0bb0
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      tools/llama/generate.py

+ 4 - 1
tools/llama/generate.py

@@ -19,7 +19,10 @@ from fish_speech.text.parser import clean_text
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 torch._inductor.config.coordinate_descent_tuning = True
 torch._inductor.config.triton.unique_kernel_names = True
-torch._inductor.config.fx_graph_cache = True  # Experimental feature to reduce compilation times, will be on by default in future
+
+if hasattr(torch._inductor.config, "fx_graph_cache"):
+    # Experimental feature to reduce compilation times, will be on by default in future
+    torch._inductor.config.fx_graph_cache = True
 
 
 from fish_speech.models.text2semantic.llama import Transformer