Commit 1ec97b70 authored by finetuneanon's avatar finetuneanon Committed by GitHub

Update models.py

parent 5ec7a358
...@@ -273,7 +273,7 @@ class StableDiffusionModel(nn.Module): ...@@ -273,7 +273,7 @@ class StableDiffusionModel(nn.Module):
print("benchmarking") print("benchmarking")
for res in res_list: for res in res_list:
for n in [1, 2, 4]: for n in [1, 2, 4]:
for w, h in [res, (res[1], res[0])]: for w, h in [res]:
settings.width = w settings.width = w
settings.height = h settings.height = h
settings.n_samples = n settings.n_samples = n
......
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