Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ def run_lora(prompt, cfg_scale, steps, selected_repo, randomize_seed, seed, widt
|
|
91 |
seed = random.randint(0, MAX_SEED)
|
92 |
|
93 |
images = generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress)
|
94 |
-
pipe.to("
|
95 |
pipe.unload_lora_weights()
|
96 |
return images, seed
|
97 |
|
|
|
91 |
seed = random.randint(0, MAX_SEED)
|
92 |
|
93 |
images = generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress)
|
94 |
+
pipe.to("cpu")
|
95 |
pipe.unload_lora_weights()
|
96 |
return images, seed
|
97 |
|