Charbel Malo commited on
Commit
dbdf8b7
1 Parent(s): da03651

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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)