AlekseyCalvin commited on
Commit
595ba94
1 Parent(s): cebb160

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -15,8 +15,7 @@ with open('loras.json', 'r') as f:
15
  loras = json.load(f)
16
 
17
  # Initialize the base model
18
- base_model = "ariG23498/sd-3.5-merged"
19
- pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
20
 
21
  MAX_SEED = 2**32-1
22
 
@@ -119,7 +118,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
119
  )
120
  # Info blob stating what the app is running
121
  info_blob = gr.HTML(
122
- """<div id="info_blob"> SOON®'s curated LoRa Gallery & Art Manufactory Space.|Runs on Stable Diffusion 3.5 Turbo. Now testing HST-triggerable historic photo-trained LoRAs. </div>"""
123
  )
124
 
125
  # Info blob stating what the app is running
@@ -152,8 +151,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
152
  with gr.Accordion("Advanced Settings", open=True):
153
  with gr.Column():
154
  with gr.Row():
155
- cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.5, value=1.5)
156
- steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=8)
157
 
158
  with gr.Row():
159
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
 
15
  loras = json.load(f)
16
 
17
  # Initialize the base model
18
+ pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16)
 
19
 
20
  MAX_SEED = 2**32-1
21
 
 
118
  )
119
  # Info blob stating what the app is running
120
  info_blob = gr.HTML(
121
+ """<div id="info_blob"> SOON®'s curated LoRa Gallery & Art Manufactory Space.|Runs on Stable Diffusion 3.5. Now testing HST-triggerable historic photo-trained LoRAs. </div>"""
122
  )
123
 
124
  # Info blob stating what the app is running
 
151
  with gr.Accordion("Advanced Settings", open=True):
152
  with gr.Column():
153
  with gr.Row():
154
+ cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.5, value=4.5)
155
+ steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=24)
156
 
157
  with gr.Row():
158
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)