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

remove sd_local_file from instruct_pix2pix

Browse Source
This commit is contained in:
Qing 2023-01-28 21:50:55 +08:00
parent b7c51c0e72
commit 5950cc9c38
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/instruct_pix2pix.py
View File

@ -16,7 +16,7 @@ class InstructPix2Pix(DiffusionInpaintModel):
from diffusers import StableDiffusionInstructPix2PixPipeline from diffusers import StableDiffusionInstructPix2PixPipeline
fp16 = not kwargs.get('no_half', False) fp16 = not kwargs.get('no_half', False)
model_kwargs = {"local_files_only": kwargs.get('local_files_only', kwargs['sd_run_local'])} model_kwargs = {"local_files_only": kwargs.get('local_files_only', False)}
if kwargs['disable_nsfw'] or kwargs.get('cpu_offload', False): if kwargs['disable_nsfw'] or kwargs.get('cpu_offload', False):
logger.info("Disable Stable Diffusion Model NSFW checker") logger.info("Disable Stable Diffusion Model NSFW checker")
model_kwargs.update(dict( model_kwargs.update(dict(