Qwen-Image-Edit-Outpaint

Running on Zero

cbensimon HF Staff commited on 17 days ago

Commit

5adfb5e

verified ·

1 Parent(s): 10b04c2

Update optimization.py

Files changed (1) hide show

optimization.py CHANGED Viewed

@@ -13,7 +13,7 @@ from torch.utils._pytree import tree_map
 P = ParamSpec('P')
-TRANSFORMER_HIDDEN_DIM = torch.export.Dim.AUTO('hidden', min=3584, max=9727)
 TRANSFORMER_DYNAMIC_SHAPES = {
     'hidden_states': {1: TRANSFORMER_HIDDEN_DIM},

 P = ParamSpec('P')
+TRANSFORMER_HIDDEN_DIM = torch.export.Dim.AUTO(min=3584, max=9727)
 TRANSFORMER_DYNAMIC_SHAPES = {
     'hidden_states': {1: TRANSFORMER_HIDDEN_DIM},