cocktailpeanut commited on
Commit
78a1133
·
1 Parent(s): 86120ee
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,13 +9,13 @@ from huggingface_hub import hf_hub_download
9
  from controlnet_union import ControlNetModel_Union
10
  from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
11
  import devicetorch
12
- DEVICE = devicetorch.get(torch)
13
 
 
 
 
14
  def init():
15
  global pipe
16
- MODELS = {
17
- "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
18
- }
19
 
20
  config_file = hf_hub_download(
21
  "xinsir/controlnet-union-sdxl-1.0",
 
9
  from controlnet_union import ControlNetModel_Union
10
  from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
11
  import devicetorch
 
12
 
13
+ MODELS = {
14
+ "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
15
+ }
16
  def init():
17
  global pipe
18
+ DEVICE = devicetorch.get(torch)
 
 
19
 
20
  config_file = hf_hub_download(
21
  "xinsir/controlnet-union-sdxl-1.0",