LPDoctor commited on
Commit
2b8313d
·
verified ·
1 Parent(s): ed6991e

replace new model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "imagepipeline/JuggernautXL-v8"
47
 
48
  pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
49
  base_model_path,
50
  controlnet=controlnet,
51
- torch_dtype=torch.float16,
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
  )