Update app.py
Browse files
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 |
-
|
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)
|