nkubana0 commited on
Commit
989e338
Β·
1 Parent(s): 203345b

fix: use CPU-only torch, remove bitsandbytes for Spaces build

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -15,11 +15,10 @@ opencv-python-headless==4.8.1.78
15
  # ── ML / NLP ───────────────────────────────────────────────────────────────────
16
  numpy==1.26.4 # Fix for version 2.0 compatibility issues
17
  Pillow==11.3.0
18
- torch>=2.2.2 # Downgraded for Python 3.9 compatibility
19
- transformers>=4.40.0 # HuggingFace Transformers (Mistral, BERT, DeBERTa, FLAN-T5)
20
  sentencepiece>=0.1.99 # Required for T5Tokenizer
21
- accelerate>=0.27.0 # Required for device_map="auto" (Mistral)
22
- bitsandbytes>=0.42.0 # Optional: 4-bit quantisation for Mistral-7B
23
  sentence-transformers==2.3.1
24
  spacy==3.7.2
25
  nltk==3.8.1
 
15
  # ── ML / NLP ───────────────────────────────────────────────────────────────────
16
  numpy==1.26.4 # Fix for version 2.0 compatibility issues
17
  Pillow==11.3.0
18
+ torch>=2.2.2 --index-url https://download.pytorch.org/whl/cpu
19
+ transformers>=4.40.0 # HuggingFace Transformers (BERT, DeBERTa, FLAN-T5)
20
  sentencepiece>=0.1.99 # Required for T5Tokenizer
21
+ accelerate>=0.27.0
 
22
  sentence-transformers==2.3.1
23
  spacy==3.7.2
24
  nltk==3.8.1