el-el-san commited on
Commit
55f1281
·
verified ·
1 Parent(s): 2ae2578

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -15,11 +15,11 @@ import spaces
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  controlnet = ControlNetModel.from_pretrained(
18
- "2vXpSwA7/test_controlnet2/CN-anytest_v4-marged_am_dim256.safetensors",
19
- #"xinsir/controlnet-scribble-sdxl-1.0",
20
- torch_dtype=torch.float16,
21
- from_tf=False,
22
- variant="safetensors"
23
  )
24
 
25
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  controlnet = ControlNetModel.from_pretrained(
18
+ #"2vXpSwA7/test_controlnet2/CN-anytest_v4-marged_am_dim256.safetensors",
19
+ "xinsir/controlnet-scribble-sdxl-1.0",
20
+ torch_dtype=torch.float16
21
+ #from_tf=False,
22
+ #variant="safetensors"
23
  )
24
 
25
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)