Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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=
|
241 |
-
guidance_scale = gr.Slider(label="Guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=
|
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 |
|