Update Storydiffusion_node.py

free v
This commit is contained in:
smthemex
2024-06-09 19:02:45 +08:00
committed by GitHub
parent 1f703ce126
commit 97de60e1ed
+1
View File
@@ -1274,6 +1274,7 @@ class Storydiffusion_Img2Img:
for value in gen:
pass_value = value
del pass_value
image = narry_list(value)
image = torch.from_numpy(np.fromiter(image, np.dtype((np.float32, (height, width, 3)))))
prompt_array = scene_prompt