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

fix sd pad_mod to 64

Browse Source 
diffusers bug: https://github.com/huggingface/diffusers/pull/505
This commit is contained in:
Qing 2022-09-27 22:37:41 +08:00
parent 03315261e2
commit 36b078123c
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 @@ from lama_cleaner.schema import Config, SDSampler
class SD(InpaintModel): class SD(InpaintModel):
pad_mod = 32 pad_mod = 64 # current diffusers only support 64 https://github.com/huggingface/diffusers/pull/505
min_size = 512 min_size = 512
def init_model(self, device: torch.device, **kwargs): def init_model(self, device: torch.device, **kwargs):