Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
182d36a
1
Parent(s):
72cad74
Update app.py
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
82 |
with calculateDuration("Randomizing seed"):
|
83 |
if randomize_seed:
|
84 |
seed = random.randint(0, MAX_SEED)
|
85 |
-
generator = torch.Generator(device="
|
86 |
|
87 |
image = generate_image(prompt, trigger_word, steps, cfg_scale, width, height, generator, lora_scale, progress)
|
88 |
|
|
|
82 |
with calculateDuration("Randomizing seed"):
|
83 |
if randomize_seed:
|
84 |
seed = random.randint(0, MAX_SEED)
|
85 |
+
generator = torch.Generator(device="cpu").manual_seed(seed)
|
86 |
|
87 |
image = generate_image(prompt, trigger_word, steps, cfg_scale, width, height, generator, lora_scale, progress)
|
88 |
|