TalHach61 commited on
Commit
fa993f9
·
verified ·
1 Parent(s): 5573002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -69,16 +69,6 @@ controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
69
 
70
  controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
71
  pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
72
-
73
- tokenizer = T5TokenizerFast.from_pretrained(
74
- base_model, subfolder="tokenizer"
75
- )
76
- text_encoder = T5EncoderModel.from_pretrained(
77
- base_model, subfolder="text_encoder", torch_dtype=torch.float16
78
- )
79
-
80
- pipe.text_encoder = text_encoder
81
- pipe.tokenizer = tokenizer
82
  pipe.to("cuda")
83
 
84
  mode_mapping = {
@@ -230,15 +220,15 @@ with gr.Blocks(css=css) as demo:
230
  minimum=0,
231
  maximum=MAX_SEED,
232
  step=1,
233
- value=42,
234
  )
235
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
236
 
237
  with gr.Accordion("Advanced settings", open=False):
238
  with gr.Column():
239
  with gr.Row():
240
- inference_steps = gr.Slider(label="Inference steps", minimum=1, maximum=50, step=1, value=24)
241
- guidance_scale = gr.Slider(label="Guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=3.5)
242
 
243
  submit_btn = gr.Button("Submit")
244
 
 
69
 
70
  controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
71
  pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
 
 
 
 
 
 
 
 
 
 
72
  pipe.to("cuda")
73
 
74
  mode_mapping = {
 
220
  minimum=0,
221
  maximum=MAX_SEED,
222
  step=1,
223
+ value=555,
224
  )
225
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
226
 
227
  with gr.Accordion("Advanced settings", open=False):
228
  with gr.Column():
229
  with gr.Row():
230
+ inference_steps = gr.Slider(label="Inference steps", minimum=1, maximum=50, step=1, value=50)
231
+ guidance_scale = gr.Slider(label="Guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=5.0)
232
 
233
  submit_btn = gr.Button("Submit")
234