byte-vortex commited on
Commit
db7a2a0
·
verified ·
1 Parent(s): df612c9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  gradio
2
  Pillow
3
- torch
4
  diffusers
5
  transformers
6
- accelerate # Often needed for device_map="auto" or certain pipeline features, even if not explicitly used here.
7
- safetensors # Recommended for faster and safer model loading
8
- sentencepiece # Gemma uses SentencePiece tokenizer
 
1
  gradio
2
  Pillow
3
+ torch # Ensure this is a CPU-only build if you strictly want to avoid GPU, though PyTorch typically installs both.
4
  diffusers
5
  transformers
6
+ accelerate # Often needed for from_pretrained optimizations, even on CPU
7
+ safetensors # For faster and safer model loading
8
+ sentencepiece # For Gemma's tokenizer