Fix Qwen reference image autogrow input names
This commit is contained in:
1 parent
508a3f4176
commit
7826388cde
2 files changed
+4
-3
No files matched your search
+3
-2
@@ -19,10 +19,11 @@ class ImageTestTests(unittest.TestCase):
|
||||
p=dict(width=512,height=512,steps=4,guidance=1)
|
||||
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']['image_1'],['21',0])
|
||||
self.assertEqual(graph['4']['inputs']['image_2'],['22',0])
|
||||
self.assertEqual(graph['4']['inputs']['images.image_1'],['21',0])
|
||||
self.assertEqual(graph['4']['inputs']['images.image_2'],['22',0])
|
||||
self.assertEqual(graph['6']['inputs']['latent_image'],['4',2])
|
||||
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'])
|
||||
def test_flux_references_feed_both_conditioning_branches(self):
|
||||
p=dict(width=512,height=512,steps=4,guidance=1)
|
||||
|
||||
Reference in new issue
Block a user