remove local_files_only from from_single_file

This commit is contained in:
Qing 2024-02-06 09:52:20 +08:00
parent 7f5333acf6
commit 97d51d3da6

View File

@ -59,7 +59,6 @@ def get_sd_model_type(model_abs_path: str) -> ModelType:
StableDiffusionInpaintPipeline.from_single_file( StableDiffusionInpaintPipeline.from_single_file(
model_abs_path, model_abs_path,
load_safety_checker=False, load_safety_checker=False,
local_files_only=True,
num_in_channels=9, num_in_channels=9,
config_files=get_config_files(), config_files=get_config_files(),
) )
@ -84,7 +83,6 @@ def get_sdxl_model_type(model_abs_path: str) -> ModelType:
model = StableDiffusionXLInpaintPipeline.from_single_file( model = StableDiffusionXLInpaintPipeline.from_single_file(
model_abs_path, model_abs_path,
load_safety_checker=False, load_safety_checker=False,
local_files_only=True,
num_in_channels=9, num_in_channels=9,
config_files=get_config_files(), config_files=get_config_files(),
) )