Spaces:
Configuration error
Configuration 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 |
)
|