From 99c7cf475ceb0692b0259c98905ef6f5ea312173 Mon Sep 17 00:00:00 2001 From: ATMxsp01 Date: Mon, 16 Dec 2024 14:52:41 +0800 Subject: [PATCH] format update --- python/llm/example/GPU/HuggingFace/LLM/chatglm3/streamchat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/example/GPU/HuggingFace/LLM/chatglm3/streamchat.py b/python/llm/example/GPU/HuggingFace/LLM/chatglm3/streamchat.py index a1f0b61352b..305e3c051f7 100644 --- 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,