Spaces:

MegaTronX
/

MetArtLoRA

Sleeping

MegaTronX commited on 15 days ago

Commit

2c447b6

verified ·

1 Parent(s): 3257204

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ pipe = DiffusionPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16
 ).to(device)
-pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors").to(device)
 @spaces.GPU(duration=75)
 def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):

     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16
 ).to(device)
+pipe.load_lora_weights("MegaTronX/MetartLoRA", weight_name="MetartLoRA.safetensors")
 @spaces.GPU(duration=75)
 def generate_image(prompt, num_inference_steps=25, guidance_scale=7.5, seed=None):