File tree Expand file tree Collapse file tree 3 files changed +4
-3
lines changed Expand file tree Collapse file tree 3 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 1
1
transformers == 4.52.4
2
2
accelerate
3
- nvfuser-cu128-torch27
3
+ nvfuser-cu128-torch28
4
4
nvidia-cudnn-frontend
5
5
einops
6
6
tiktoken
Original file line number Diff line number Diff line change 1
1
transformers == 4.55.4
2
2
accelerate
3
- nvfuser-cu128-torch27
3
+ nvfuser-cu128-torch28
4
4
nvidia-cudnn-frontend
Original file line number Diff line number Diff line change @@ -34,10 +34,11 @@ def get_nvfuser_package_hint() -> str:
34
34
else :
35
35
return f"""nvFuser was specified but we don't currently support torch { torch_version } .
36
36
37
- Please upgrade to torch 2.6 or 2.7 and then run
37
+ Please upgrade to torch 2.6 to 2.8 and then run
38
38
```
39
39
pip install nvfuser-cu126-torch26 # for torch 2.6
40
40
pip install nvfuser-cu128-torch27 # for torch 2.7
41
+ pip install nvfuser-cu128-torch28 # for torch 2.8
41
42
```
42
43
43
44
For compatibility options, see:
You can’t perform that action at this time.
0 commit comments