Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|