Commit 5ec7a358 authored by finetuneanon's avatar finetuneanon Committed by GitHub

Update models.py

parent 3c519148
...@@ -272,8 +272,8 @@ class StableDiffusionModel(nn.Module): ...@@ -272,8 +272,8 @@ class StableDiffusionModel(nn.Module):
results = [] results = []
print("benchmarking") print("benchmarking")
for res in res_list: for res in res_list:
for w, h in [res, (res[1], res[0])]: for n in [1, 2, 4]:
for n in [1, 2, 4]: for w, h in [res, (res[1], res[0])]:
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