Spaces:

TotoB12
/

Fast-Flux

Sleeping

TotoB12 commited on Dec 7, 2024

Commit

d476216

verified ·

1 Parent(s): a29d9b4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from diffusers import DiffusionPipeline
 from transformers import pipeline
-pipe = pipeline(model="black-forest-labs/FLUX.1-schnell")
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:

 from transformers import pipeline
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed: