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

确定精度

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ base_model_path = "stablediffusionapi/juggernaut-xl-v8"
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
  )
 
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
  )