Spaces:
Runtime error
Runtime error
add token auth
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ MONSTER_V2 = "monster-labs/V2"
|
|
18 |
LANDMARKS = "monster-labs/V2"
|
19 |
|
20 |
controlnet = [
|
21 |
-
ControlNetModel.from_pretrained(MONSTER_V2, torch_dtype=torch.float16, subfolder="step1"),
|
22 |
-
ControlNetModel.from_pretrained(LANDMARKS, torch_dtype=torch.float16, subfolder="step2"),
|
23 |
]
|
24 |
|
25 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
|
18 |
LANDMARKS = "monster-labs/V2"
|
19 |
|
20 |
controlnet = [
|
21 |
+
ControlNetModel.from_pretrained(MONSTER_V2, torch_dtype=torch.float16, subfolder="step1", use_auth_token=os.environ["HUGGINGFACE_TOKEN"]),
|
22 |
+
ControlNetModel.from_pretrained(LANDMARKS, torch_dtype=torch.float16, subfolder="step2", use_auth_token=os.environ["HUGGINGFACE_TOKEN"]),
|
23 |
]
|
24 |
|
25 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|