anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Refactor][Diffusion] Remove model-specific names from LoRA and ModelOpt loader defaults (#5907)

Browse Source 
Signed-off-by: Alicia <115451386+congw729@users.noreply.github.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
AliciaandHongsheng Liu authored and GitHub committed 2026-09-12 20:30:03 +08:00
1 parent bc0c9f4b45
commit 28291df179
6 files changed
+43 -12
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/diffusion/models/bagel/test_bagel_lora.py
+7 -2
View File
@@ -122,8 +122,10 @@ class TestStage1DiTLoRA:
lambda root, name, sub: fake_replace_submodule(root, name, sub, replace_calls),
)
# Build pipeline with bagel component
# Build pipeline with bagel component declared via _lora_components,
# matching how BagelPipeline opts in.
pipeline = torch.nn.Module()
pipeline._lora_components = ["bagel"]
pipeline.bagel = torch.nn.Module()
lm = torch.nn.Module()
lm.stacked_params_mapping = [
@@ -194,8 +196,11 @@ class TestBagelLoRARoundTrip:
monkeypatch.setattr(manager_mod, "BaseLayerWithLoRA", DummyBaseLayerWithLoRA)
# Build pipeline with bagel.language_model.foo (simple non-packed layer)
# Build pipeline with bagel.language_model.foo (simple non-packed layer);
# the bagel component is declared via _lora_components, matching how
# BagelPipeline opts in.
pipeline = torch.nn.Module()
pipeline._lora_components = ["bagel"]
pipeline.bagel = torch.nn.Module()
lm = torch.nn.Module()
lm.foo = _FakeLinearBase()