Update app.py
Browse files
app.py
CHANGED
@@ -86,6 +86,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
86 |
seed = random.randint(0, MAX_SEED)
|
87 |
|
88 |
image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
|
|
|
89 |
pipe.to("cuda")
|
90 |
pipe.unload_lora_weights()
|
91 |
return image, seed
|
|
|
86 |
seed = random.randint(0, MAX_SEED)
|
87 |
|
88 |
image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
|
89 |
+
print("Model : " + selected_lora["title"] + " Prompt : " + prompt)
|
90 |
pipe.to("cuda")
|
91 |
pipe.unload_lora_weights()
|
92 |
return image, seed
|