linoyts HF staff commited on
Commit
8f2f81c
·
verified ·
1 Parent(s): 5a84df5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -19,8 +19,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
19
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
20
  custom_pipeline="pipeline_flux_rf_inversion",
21
  torch_dtype=torch.bfloat16)
22
- pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
23
- pipe.fuse_lora(lora_scale=0.125)
24
  #pipe.enable_lora()
25
  pipe.to(DEVICE)
26
 
@@ -44,6 +43,7 @@ def check_style(stylezation):
44
 
45
  def check_hyper_flux_lora(enable_hyper_flux):
46
  if enable_hyper_flux:
 
47
  pipe.fuse_lora(lora_scale=0.125)
48
  return 8, 8, 4
49
  else:
@@ -148,7 +148,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
148
  minimum=0,
149
  maximum=28,
150
  step=1,
151
- value=4,
152
  )
153
  eta = gr.Slider(
154
  label="eta",
@@ -181,7 +181,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
181
  minimum=1,
182
  maximum=50,
183
  step=1,
184
- value=8,
185
  )
186
  eta_decay = gr.Checkbox(label="eta decay", value=False)
187
  decay_power = gr.Slider(
@@ -206,7 +206,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
206
  minimum=1,
207
  maximum=50,
208
  step=1,
209
- value=8,
210
  )
211
 
212
  with gr.Row():
 
19
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
20
  custom_pipeline="pipeline_flux_rf_inversion",
21
  torch_dtype=torch.bfloat16)
22
+
 
23
  #pipe.enable_lora()
24
  pipe.to(DEVICE)
25
 
 
43
 
44
  def check_hyper_flux_lora(enable_hyper_flux):
45
  if enable_hyper_flux:
46
+ pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
47
  pipe.fuse_lora(lora_scale=0.125)
48
  return 8, 8, 4
49
  else:
 
148
  minimum=0,
149
  maximum=28,
150
  step=1,
151
+ value=6,
152
  )
153
  eta = gr.Slider(
154
  label="eta",
 
181
  minimum=1,
182
  maximum=50,
183
  step=1,
184
+ value=28,
185
  )
186
  eta_decay = gr.Checkbox(label="eta decay", value=False)
187
  decay_power = gr.Slider(
 
206
  minimum=1,
207
  maximum=50,
208
  step=1,
209
+ value=28,
210
  )
211
 
212
  with gr.Row():