AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
Fix missing variable assign

lol
This commit is contained in:
Mylo
2023-05-03 18:31:04 +02:00
committed by Daniel Gu
parent d38b4d9555
commit 0d2206459e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py
View File

@@ -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: