[diffusion] feat: support LoRA (#13859)

This commit is contained in:
Mick
2025-11-26 00:21:33 +08:00
committed by GitHub
parent 46673b4224
commit dfd7ab9682
29 changed files with 463 additions and 278 deletions

View File

@@ -621,6 +621,19 @@ class ServerArgs:
default=ServerArgs.override_transformer_cls_name,
help="Override transformer cls name",
)
# LoRA
parser.add_argument(
"--lora-path",
type=str,
default=ServerArgs.lora_path,
help="The path to the LoRA adapter weights (can be local file path or HF hub id) to launch with",
)
parser.add_argument(
"--lora-nickname",
type=str,
default=ServerArgs.lora_nickname,
help="The nickname for the LoRA adapter to launch with",
)
# Add pipeline configuration arguments
PipelineConfig.add_cli_args(parser)