michael/Athena-Deck
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Qwen reference image autogrow input names

Browse Source
This commit is contained in:
Mikei386
2026-10-01 14:06:59 +02:00
parent 508a3f4176
commit 7826388cde
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
image_test.py
+1 -1
View File
@@ -93,7 +93,7 @@ def workflow(prompt,params,seed,family='qwen',references=()):
graph['4']['inputs']['vae']=['3',0] graph['4']['inputs']['vae']=['3',0]
for index,name in enumerate(references,1): for index,name in enumerate(references,1):
graph[str(20+index)]={'class_type':'LoadImage','inputs':{'image':name}} graph[str(20+index)]={'class_type':'LoadImage','inputs':{'image':name}}
graph['4']['inputs'][f'image_{index}']=[str(20+index),0] graph['4']['inputs'][f'images.image_{index}']=[str(20+index),0]
graph['6']['inputs']['latent_image']=['4',2] graph['6']['inputs']['latent_image']=['4',2]
return graph return graph
test_image_test.py
+3 -2
View File
@@ -19,10 +19,11 @@ class ImageTestTests(unittest.TestCase):
p=dict(width=512,height=512,steps=4,guidance=1) p=dict(width=512,height=512,steps=4,guidance=1)
graph=workflow('synthetic edit',p,42,references=['reference-1.png','reference-2.jpg']) graph=workflow('synthetic edit',p,42,references=['reference-1.png','reference-2.jpg'])
self.assertEqual(graph['4']['inputs']['vae'],['3',0]) self.assertEqual(graph['4']['inputs']['vae'],['3',0])
self.assertEqual(graph['4']['inputs']['image_1'],['21',0]) self.assertEqual(graph['4']['inputs']['images.image_1'],['21',0])
self.assertEqual(graph['4']['inputs']['image_2'],['22',0]) self.assertEqual(graph['4']['inputs']['images.image_2'],['22',0])
self.assertEqual(graph['6']['inputs']['latent_image'],['4',2]) self.assertEqual(graph['6']['inputs']['latent_image'],['4',2])
self.assertEqual(graph['21']['inputs']['image'],'reference-1.png') self.assertEqual(graph['21']['inputs']['image'],'reference-1.png')
self.assertNotIn('image_1',graph['4']['inputs'])
with self.assertRaises(ValueError):workflow('test',p,1,'unknown',['reference-1.png']) with self.assertRaises(ValueError):workflow('test',p,1,'unknown',['reference-1.png'])
def test_flux_references_feed_both_conditioning_branches(self): def test_flux_references_feed_both_conditioning_branches(self):
p=dict(width=512,height=512,steps=4,guidance=1) p=dict(width=512,height=512,steps=4,guidance=1)