Passing in custom arguments
from auto1111sdk import StableDiffusionPipeline
pipe = StableDiffusionPipeline("model.safetensors", default_command_args = "--medvram")from auto1111sdk import StableDiffusionPipeline, EsrganPipeline, RealEsrganPipeline, StableDiffusionXLPipeline
pipe = StableDiffusionXLPipeline("model.safetensors", default_command_args = "--medvram")
upscaler_1 = EsrganPipeline("upscaler.pth", default_command_args = "<anything flag you want here>")
upscaler_1 = RealEsrganPipeline("upscaler2.pth", default_command_args = "<anything flag you want here>")
## For StableDiffusionPipeline, EsrganPipeline, RealEsrganPipeline
if default_command_args is None:
if torch.cuda.is_available():
os.environ['COMMANDLINE_ARGS'] = "--upcast-sampling --skip-torch-cuda-test --no-half-vae interrogate"
elif torch.backends.mps.is_available():
os.environ['COMMANDLINE_ARGS'] = "--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate"
else:
os.environ['COMMANDLINE_ARGS'] = "--skip-torch-cuda-test --no-half-vae --no-half interrogate"
else:
os.environ['COMMANDLINE_ARGS'] = default_command_argsLast updated