From 3137a465ab365e717a35a87427b83c35f2584eb4 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 15 Sep 2023 06:56:01 -0500 Subject: [PATCH] adjust node names in projection test --- api/tests/convert/diffusion/test_lora.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/tests/convert/diffusion/test_lora.py b/api/tests/convert/diffusion/test_lora.py index cbc64979..58462b80 100644 --- a/api/tests/convert/diffusion/test_lora.py +++ b/api/tests/convert/diffusion/test_lora.py @@ -126,14 +126,14 @@ class FixXLNameTests(unittest.TestCase): 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"], })