Qwen-Image-Light-Migration

Running on Zero

multimodalart HF Staff commited on 5 days ago

Commit

8e36388

verified ·

1 Parent(s): 836f311

Update optimization.py

Files changed (1) hide show

optimization.py CHANGED Viewed

@@ -27,11 +27,11 @@ TRANSFORMER_DYNAMIC_SHAPES = {
     'encoder_hidden_states_mask': {
         1: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
     },
-    'image_rotary_emb': ({
-        0: TRANSFORMER_IMAGE_SEQ_LENGTH_DIM,
-    }, {
-        0: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
-    }),
 }

     'encoder_hidden_states_mask': {
         1: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
     },
+    #'image_rotary_emb': ({
+    #    0: TRANSFORMER_IMAGE_SEQ_LENGTH_DIM,
+    #}, {
+    #    0: TRANSFORMER_TEXT_SEQ_LENGTH_DIM,
+    #}),
 }