kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

fix sd --no-half

Browse Source
This commit is contained in:
Qing 2023-01-04 21:27:37 +08:00
parent 8099841dc8
commit d2cec0d415
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/model/sd.py
View File

@ -40,7 +40,7 @@ class SD(InpaintModel):
)) ))
use_gpu = device == torch.device('cuda') and torch.cuda.is_available() use_gpu = device == torch.device('cuda') and torch.cuda.is_available()
torch_dtype = torch.float16 if use_gpu else torch.float32 torch_dtype = torch.float16 if use_gpu and fp16 else torch.float32
self.model = StableDiffusionInpaintPipeline.from_pretrained( self.model = StableDiffusionInpaintPipeline.from_pretrained(
self.model_id_or_path, self.model_id_or_path,
revision="fp16" if use_gpu and fp16 else "main", revision="fp16" if use_gpu and fp16 else "main",