Spaces:
Runtime error
Runtime error
replace new model
Browse files
app.py
CHANGED
@@ -43,12 +43,12 @@ controlnet_path = "./checkpoints/ControlNetModel"
|
|
43 |
# Load pipeline
|
44 |
controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
|
45 |
|
46 |
-
base_model_path = "
|
47 |
|
48 |
pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
|
49 |
base_model_path,
|
50 |
controlnet=controlnet,
|
51 |
-
torch_dtype=torch.
|
52 |
safety_checker=None,
|
53 |
feature_extractor=None,
|
54 |
)
|
|
|
43 |
# Load pipeline
|
44 |
controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
|
45 |
|
46 |
+
base_model_path = "stablediffusionapi/juggernaut-xl-v8"
|
47 |
|
48 |
pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
|
49 |
base_model_path,
|
50 |
controlnet=controlnet,
|
51 |
+
torch_dtype=torch.dtype,
|
52 |
safety_checker=None,
|
53 |
feature_extractor=None,
|
54 |
)
|