jbilcke-hf HF staff commited on
Commit
b1855df
·
1 Parent(s): 0d34ea8

upgrade torchao

Browse files
requirements.txt CHANGED
@@ -6,7 +6,7 @@ numpy>=1.26.4
6
  # on some system (Python 3.13+) those do not work:
7
  torch==2.5.1
8
  torchvision==0.20.1
9
- torchao==0.6.1
10
 
11
  # datasets 3.4.0 replaces decord by torchvision
12
  # let's free it for now
 
6
  # on some system (Python 3.13+) those do not work:
7
  torch==2.5.1
8
  torchvision==0.20.1
9
+ torchao>=0.7.0
10
 
11
  # datasets 3.4.0 replaces decord by torchvision
12
  # let's free it for now
requirements_without_flash_attention.txt CHANGED
@@ -6,7 +6,7 @@ numpy>=1.26.4
6
  # on some system (Python 3.13+) those do not work:
7
  torch==2.5.1
8
  torchvision==0.20.1
9
- torchao==0.6.1
10
 
11
  # datasets 3.4.0 replaces decord by torchvision
12
  # let's free it for now
 
6
  # on some system (Python 3.13+) those do not work:
7
  torch==2.5.1
8
  torchvision==0.20.1
9
+ torchao>=0.7.0
10
 
11
  # datasets 3.4.0 replaces decord by torchvision
12
  # let's free it for now