Hugging Face's logo

Spaces:
freddyaboulton
/
FLUX.1-Kontext-Dev
Running on Zero

App Files Community
freddyaboulton HF Staff commited on
Commit
8cccd14
·
verified ·
1 Parent(s): 95791e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from diffusers.utils import load_image
10
 
11
  MAX_SEED = np.iinfo(np.int32).max
12
 
13
- pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16)
14
 
15
  @spaces.GPU
16
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
 
10
 
11
  MAX_SEED = np.iinfo(np.int32).max
12
 
13
+ pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
14
 
15
  @spaces.GPU
16
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):