frogleo commited on
Commit
b42cbda
·
verified ·
1 Parent(s): 61c16d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -13,12 +13,9 @@ from PIL import Image
13
  from diffusers import FluxKontextPipeline
14
  from diffusers.utils import load_image
15
 
16
- from optimization import optimize_pipeline_
17
-
18
  MAX_SEED = np.iinfo(np.int32).max
19
 
20
  pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
21
- optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')
22
 
23
  @spaces.GPU
24
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress()):
 
13
  from diffusers import FluxKontextPipeline
14
  from diffusers.utils import load_image
15
 
 
 
16
  MAX_SEED = np.iinfo(np.int32).max
17
 
18
  pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
19
 
20
  @spaces.GPU
21
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress()):