6Morpheus6 commited on
Commit
d9f69cc
·
verified ·
1 Parent(s): f00ffec

load model from non-gated repo

Files changed (1) hide show
  1. app.py +3 -3
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).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)):
@@ -106,7 +106,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
106
  container=False,
107
  )
108
  run_button = gr.Button("Run", scale=0)
109
- with gr.Accordion("Advanced Settings", open=False):
110
 
111
  seed = gr.Slider(
112
  label="Seed",
@@ -148,7 +148,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
148
  inputs=[input_image, prompt],
149
  outputs=[result, seed],
150
  fn=infer_example,
151
- cache_examples="lazy"
152
  )
153
 
154
  gr.on(
 
10
 
11
  MAX_SEED = np.iinfo(np.int32).max
12
 
13
+ pipe = FluxKontextPipeline.from_pretrained("fuliucansheng/FLUX.1-Kontext-dev-diffusers", 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)):
 
106
  container=False,
107
  )
108
  run_button = gr.Button("Run", scale=0)
109
+ with gr.Accordion("Advanced Settings", open=True):
110
 
111
  seed = gr.Slider(
112
  label="Seed",
 
148
  inputs=[input_image, prompt],
149
  outputs=[result, seed],
150
  fn=infer_example,
151
+ cache_examples="False"
152
  )
153
 
154
  gr.on(