aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/LDSR/ldsr_model_arch.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-10 08:37:18 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-05-10 08:37:18 +0000
commita5121e7a0623db328a9462d340d389ed6737374a (patch)
tree2fa6051d457555ef2d61793af2756a44c0ea221c /extensions-builtin/LDSR/ldsr_model_arch.py
parent550256db1ce18778a9d56ff343d844c61b9f9b83 (diff)
downloadstable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.tar.gz
stable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.tar.bz2
stable-diffusion-webui-gfx803-a5121e7a0623db328a9462d340d389ed6737374a.zip
fixes for B007
Diffstat (limited to 'extensions-builtin/LDSR/ldsr_model_arch.py')
-rw-r--r--extensions-builtin/LDSR/ldsr_model_arch.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py
index a5fb8907..27e38549 100644
--- a/extensions-builtin/LDSR/ldsr_model_arch.py
+++ b/extensions-builtin/LDSR/ldsr_model_arch.py
@@ -88,7 +88,7 @@ class LDSR:
x_t = None
logs = None
- for n in range(n_runs):
+ for _ in range(n_runs):
if custom_shape is not None:
x_t = torch.randn(1, custom_shape[1], custom_shape[2], custom_shape[3]).to(model.device)
x_t = repeat(x_t, '1 c h w -> b c h w', b=custom_shape[0])