fix tensor type

zhtmike · zhtmike · commit 90c5ede46e0d · 2025-10-15T13:37:33.000+08:00
diff --git a/examples/wan2_1/wan/utils/prompt_extend.py b/examples/wan2_1/wan/utils/prompt_extend.py
@@ -252,7 +252,7 @@ def extend_with_img(self, prompt, system_prompt, image: Union[Image.Image, str]
             images=image_inputs,
             videos=video_inputs,
             padding=True,
-            return_tensors="pt",
+            return_tensors="np",
         )
 
         for k, v in inputs.items():

Original file line number	Diff line number	Diff line change
`@@ -252,7 +252,7 @@ def extend_with_img(self, prompt, system_prompt, image: Union[Image.Image, str]`
`252`	`252`	`images=image_inputs,`
`253`	`253`	`videos=video_inputs,`
`254`	`254`	`padding=True,`
`255`		`- return_tensors="pt",`
	`255`	`+ return_tensors="np",`
`256`	`256`	`)`
`257`	`257`
`258`	`258`	`for k, v in inputs.items():`