Spaces:
Running
on
Zero
Running
on
Zero
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 |
|