Unverified Commit 63a8ef7b authored by Mylo's avatar Mylo Committed by GitHub
Browse files

Fix missing variable assign in DeepFloyd-IF-II (#3315)

Fix missing variable assign

lol
parent 0ccad2ad
......@@ -667,7 +667,7 @@ class IFSuperResolutionPipeline(DiffusionPipeline):
image = [np.array(i).astype(np.float32) / 255.0 for i in image]
image = np.stack(image, axis=0) # to np
torch.from_numpy(image.transpose(0, 3, 1, 2))
image = torch.from_numpy(image.transpose(0, 3, 1, 2))
elif isinstance(image[0], np.ndarray):
image = np.stack(image, axis=0) # to np
if image.ndim == 5:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment