pcuenq HF staff kashif HF staff commited on
Commit
50e4dde
1 Parent(s): 455bbbc

update requirements to install cpu only torch (#8)

Browse files

- update requirements to install cpu only torch (28a599a576f903a8bc68c554d77ab797b5f25155)


Co-authored-by: Kashif Rasul <kashif@users.noreply.huggingface.co>

Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
  git+https://github.com/huggingface/huggingface_hub@main
2
  git+https://github.com/huggingface/diffusers.git
 
3
  torch
4
  #transformers
5
  git+https://github.com/huggingface/transformers
6
- pytorch_lightning
7
  OmegaConf
8
  ftfy
9
  safetensors
 
1
  git+https://github.com/huggingface/huggingface_hub@main
2
  git+https://github.com/huggingface/diffusers.git
3
+ --find-links https://download.pytorch.org/whl/cpu
4
  torch
5
  #transformers
6
  git+https://github.com/huggingface/transformers
7
+ lightning
8
  OmegaConf
9
  ftfy
10
  safetensors