Liangyingping commited on
Commit
76e4b82
·
verified ·
1 Parent(s): bc6fa33

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ model_path = "Liangyingping/L2M-Inpainting" # 替换为你自己的模型路径
11
 
12
  # 加载模型
13
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
14
- model_path, torch_dtype=torch.float16, local_files_only=True
15
  )
16
  pipe.to("cuda") # 如果有 GPU,可以将模型加载到 GPU 上
17
 
 
11
 
12
  # 加载模型
13
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
14
+ model_path, torch_dtype=torch.float16
15
  )
16
  pipe.to("cuda") # 如果有 GPU,可以将模型加载到 GPU 上
17