format update

intel-analytics · Dec 16, 2024 · 99c7cf4 · 99c7cf4
1 parent 696283c
commit 99c7cf4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/llm/example/GPU/HuggingFace/LLM/chatglm3/streamchat.py b/python/llm/example/GPU/HuggingFace/LLM/chatglm3/streamchat.py
@@ -57,7 +57,7 @@
                                       trust_remote_code=True,
                                       optimize_model=True,
                                       model_hub=model_hub)
-    model.half().to('xpu')
+    model = model.half().to('xpu')
 
     # Load tokenizer
     tokenizer = AutoTokenizer.from_pretrained(model_path,