adjust node names in projection test

ssube · Sep 15, 2023 · 3137a46 · 3137a46
1 parent b851c23
commit 3137a46
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/tests/convert/diffusion/test_lora.py b/api/tests/convert/diffusion/test_lora.py
@@ -126,14 +126,14 @@ def test_unmatched_block(self):
 
     def test_output_projection(self):
         nodes = {
-            "output_block_proj_o.lora_down.weight": {},
+            "output_block_proj_out.lora_down.weight": {},
         }
         fixed = fix_xl_names(nodes, [
-            NodeProto(name="/up_blocks_proj_o/MatMul"),
+            NodeProto(name="/up_blocks_proj_out/MatMul"),
         ])
 
         self.assertEqual(fixed, {
-            "up_blocks_proj_out": nodes["output_block_proj_o.lora_down.weight"],
+            "up_blocks_proj_out": nodes["output_block_proj_out.lora_down.weight"],
         })