guardiancc commited on
Commit
17c4e15
1 Parent(s): 061cbe1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -30,7 +30,6 @@ state_dict = load_state_dict(model_file)
30
  model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
31
  controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
32
  )
33
- model.to(device="cpu", dtype=torch.float16)
34
 
35
  vae = AutoencoderKL.from_pretrained(
36
  "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
@@ -276,15 +275,6 @@ with gr.Blocks(css=css) as demo:
276
 
277
  run_button = gr.Button("Generate")
278
 
279
- gr.Examples(
280
- examples = [
281
- ["./examples/example_1.webp", "RealVisXL V5.0 Lightning", "16:9"],
282
- ["./examples/example_2.jpg", "RealVisXL V5.0 Lightning", "16:9"],
283
- ["./examples/example_3.jpg", "RealVisXL V5.0 Lightning", "9:16"]
284
- ],
285
- inputs = [input_image, model_selection, ratio]
286
- )
287
-
288
  with gr.Column():
289
  result = ImageSlider(
290
  interactive=False,
@@ -302,4 +292,4 @@ with gr.Blocks(css=css) as demo:
302
  )
303
 
304
 
305
- demo.launch(share=False)
 
30
  model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
31
  controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
32
  )
 
33
 
34
  vae = AutoencoderKL.from_pretrained(
35
  "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
 
275
 
276
  run_button = gr.Button("Generate")
277
 
 
 
 
 
 
 
 
 
 
278
  with gr.Column():
279
  result = ImageSlider(
280
  interactive=False,
 
292
  )
293
 
294
 
295
+ demo.launch()