adjust node names in projection test
This commit is contained in:
parent
b851c234fe
commit
3137a465ab
|
@ -126,14 +126,14 @@ class FixXLNameTests(unittest.TestCase):
|
||||||
|
|
||||||
def test_output_projection(self):
|
def test_output_projection(self):
|
||||||
nodes = {
|
nodes = {
|
||||||
"output_block_proj_o.lora_down.weight": {},
|
"output_block_proj_out.lora_down.weight": {},
|
||||||
}
|
}
|
||||||
fixed = fix_xl_names(nodes, [
|
fixed = fix_xl_names(nodes, [
|
||||||
NodeProto(name="/up_blocks_proj_o/MatMul"),
|
NodeProto(name="/up_blocks_proj_out/MatMul"),
|
||||||
])
|
])
|
||||||
|
|
||||||
self.assertEqual(fixed, {
|
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"],
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue