little polish in reference info and indent

intel-analytics · Dec 24, 2024 · 458c59b · 458c59b
1 parent a171de0
commit 458c59b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/llm/example/GPU/HuggingFace/Multimodal/glm-edge-v/generate.py b/python/llm/example/GPU/HuggingFace/Multimodal/glm-edge-v/generate.py
@@ -79,7 +79,7 @@
     pixel_values = pixel_values.to('xpu')
 
     with torch.inference_mode():
-        # The following code for generation is adapted from https://huggingface.co/THUDM/glm-edge-v-2b#inference
+        # The following code for generation is adapted from https://huggingface.co/THUDM/glm-edge-v-5b#inference
         messages = [{
             "role": "user", 
             "content": [{"type": "image"}, 
@@ -92,8 +92,8 @@
                                 add_generation_prompt=True, 
                                 return_dict=True, 
                                 tokenize=True, 
-                                return_tensors="pt"
-                            ).to('xpu')
+                                return_tensors="pt")
+        inputs = inputs.to('xpu')
 
         generate_kwargs = {
                             **inputs,