AlekseyCalvin commited on
Commit
678d7db
·
verified ·
1 Parent(s): 632bc08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -50,7 +50,7 @@ pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF1
50
 
51
  device = "cuda" if torch.cuda.is_available() else "cpu"
52
 
53
- pipe.enable_model_cpu_offload()
54
 
55
  try: # A temp hack for some version diffusers lora loading problem
56
  from diffusers.utils.peft_utils import _derive_exclude_modules
@@ -207,7 +207,7 @@ with gr.Blocks(css=css) as app:
207
  with gr.Column():
208
  with gr.Row():
209
  cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.1, value=1.0)
210
- steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=9)
211
 
212
  with gr.Row():
213
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
@@ -216,7 +216,7 @@ with gr.Blocks(css=css) as app:
216
  with gr.Row():
217
  randomize_seed = gr.Checkbox(True, label="Randomize seed")
218
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
219
- lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.5, step=0.01, value=1.0)
220
 
221
  gallery.select(
222
  update_selection,
 
50
 
51
  device = "cuda" if torch.cuda.is_available() else "cpu"
52
 
53
+ #pipe.enable_model_cpu_offload()
54
 
55
  try: # A temp hack for some version diffusers lora loading problem
56
  from diffusers.utils.peft_utils import _derive_exclude_modules
 
207
  with gr.Column():
208
  with gr.Row():
209
  cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.1, value=1.0)
210
+ steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=10)
211
 
212
  with gr.Row():
213
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
 
216
  with gr.Row():
217
  randomize_seed = gr.Checkbox(True, label="Randomize seed")
218
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
219
+ lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.5, step=0.01, value=0.8)
220
 
221
  gallery.select(
222
  update_selection,