Spaces:
Running
Running
Charbel Malo
commited on
Commit
•
dbdf8b7
1
Parent(s):
da03651
Update app.py
Browse files
app.py
CHANGED
@@ -270,7 +270,6 @@ def remove_custom_lora(selected_indices, current_loras, gallery):
|
|
270 |
lora_image_2
|
271 |
)
|
272 |
|
273 |
-
@spaces.GPU()
|
274 |
def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
|
275 |
print("Generating image...")
|
276 |
pipe.to("cuda")
|
@@ -290,7 +289,6 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
|
|
290 |
):
|
291 |
yield img
|
292 |
|
293 |
-
@spaces.GPU()
|
294 |
def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
|
295 |
pipe_i2i.to("cuda")
|
296 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
|
|
270 |
lora_image_2
|
271 |
)
|
272 |
|
|
|
273 |
def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
|
274 |
print("Generating image...")
|
275 |
pipe.to("cuda")
|
|
|
289 |
):
|
290 |
yield img
|
291 |
|
|
|
292 |
def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
|
293 |
pipe_i2i.to("cuda")
|
294 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|