Skip to content

Commit a0803f9

Browse files
committed
fix order
1 parent 0580379 commit a0803f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/sana/test_sana_sprint_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,11 +133,11 @@ def get_dummy_components(self):
133133
return components
134134

135135
def get_dummy_inputs(self, device, seed=0):
136-
image = torch.randn(1, 3, 32, 32, generator=generator)
137136
if str(device).startswith("mps"):
138137
generator = torch.manual_seed(seed)
139138
else:
140139
generator = torch.Generator(device=device).manual_seed(seed)
140+
image = torch.randn(1, 3, 32, 32, generator=generator)
141141
inputs = {
142142
"prompt": "",
143143
"image": image,

0 commit comments

Comments
 (0)