[Kernel] Add punica dimension for Swallow-MS-7B LoRA (#4134)

2024-04-18 02:02:45 +09:00
parent fe3b5bbc23
commit a53222544c
2 changed files with 2 additions and 0 deletions
--- a/tests/lora/test_punica.py
+++ b/tests/lora/test_punica.py
@@ -82,6 +82,7 @@ H1 = H2 = [
    32768,
    33024,
    36864,
+    43264,
    49152,
    64000,
    64256,