support OFT

okotaku · Dec 7, 2023 · ac0d083 · ac0d083
1 parent c32af46
commit ac0d083
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/configs/_base_/models/stable_diffusion_xl_oft.py b/configs/_base_/models/stable_diffusion_xl_oft.py
@@ -5,5 +5,4 @@
     unet_lora_config=dict(
         type="OFT",
         r=8,
-        alpha=1,
         target_modules=["to_q", "to_v", "to_k", "to_out.0"]))
diff --git a/tests/test_models/test_archs.py b/tests/test_models/test_archs.py
@@ -57,9 +57,7 @@ def test_create_peft_config():
     config = dict(
         type="OFT",
         r=8,
-        alpha=2,
     )
     config = create_peft_config(config)
     assert isinstance(config, OFTConfig)
     assert config.r == 8
-    assert config.alpha == 2