update requirements to install cpu only torch

#8
by kashif HF staff - opened
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