eliphatfs commited on
Commit
61ba9ac
1 Parent(s): 22e326e

Update requirements for GPU.

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,6 +1,7 @@
 
1
  torch
2
  numpy
3
- dgl
4
  # open_clip_torch
5
  diffusers
6
  accelerate
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch
3
  numpy
4
+ dgl-cu113
5
  # open_clip_torch
6
  diffusers
7
  accelerate