chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] feat: support LoRA (#13859)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/test/server/test_server_common.py
+3 -4
View File
@@ -73,7 +73,7 @@ def diffusion_server(case: DiffusionTestCase) -> ServerContext:
prompt=sampling_params.prompt or "A colorful raccoon icon",
output_size=output_size,
)
warmup.run_text_warmups(case.server_args.warmup_text)
warmup.run_text_warmups(server_args.warmup_text)
if (
case.server_args.warmup_edit > 0
@@ -81,15 +81,14 @@ def diffusion_server(case: DiffusionTestCase) -> ServerContext:
and sampling_params.image_path
):
# Handle URL or local path
image_path = sampling_params.image_path
if is_image_url(sampling_params.image_path):
image_path = download_image_from_url(str(sampling_params.image_path))
else:
image_path = Path(sampling_params.image_path)
warmup.run_edit_warmups(
count=case.server_args.warmup_edit,
edit_prompt=case.sampling_params.prompt,
count=server_args.warmup_edit,
edit_prompt=sampling_params.prompt,
image_path=image_path,
)
except Exception as exc: