Hugging Face's logo

Spaces:
ginigen
/
Nano-Banana-PRO
Paused

App Files Community
1
cbensimon HF Staff commited on
Commit
5adfb5e
·
verified ·
1 Parent(s): 10b04c2

Update optimization.py

Browse files
Files changed (1) hide show
  1. optimization.py +1 -1
optimization.py CHANGED
@@ -13,7 +13,7 @@ from torch.utils._pytree import tree_map
13
  P = ParamSpec('P')
14
 
15
 
16
- TRANSFORMER_HIDDEN_DIM = torch.export.Dim.AUTO('hidden', min=3584, max=9727)
17
 
18
  TRANSFORMER_DYNAMIC_SHAPES = {
19
  'hidden_states': {1: TRANSFORMER_HIDDEN_DIM},
 
13
  P = ParamSpec('P')
14
 
15
 
16
+ TRANSFORMER_HIDDEN_DIM = torch.export.Dim.AUTO(min=3584, max=9727)
17
 
18
  TRANSFORMER_DYNAMIC_SHAPES = {
19
  'hidden_states': {1: TRANSFORMER_HIDDEN_DIM},