MegaTronX commited on
Commit
3043719
·
verified ·
1 Parent(s): 3563a40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ pipe = DiffusionPipeline.from_pretrained(
10
  pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors")
11
  pipe = pipe.to("cuda")
12
 
13
- #@spaces.GPU
14
  def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):
15
  """Generates an image using the FLUX.1-dev LoRA model."""
16
  generator = torch.Generator("cuda").manual_seed(seed) if seed else None
 
10
  pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors")
11
  pipe = pipe.to("cuda")
12
 
13
+ #@spaces.GPU(duration=75)
14
  def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):
15
  """Generates an image using the FLUX.1-dev LoRA model."""
16
  generator = torch.Generator("cuda").manual_seed(seed) if seed else None