hajimammad commited on
Commit
89bc368
·
verified ·
1 Parent(s): 9ef9347

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -24
requirements.txt CHANGED
@@ -1,38 +1,20 @@
1
- # ========= Mahoon / Training Profile =========
2
- # UI (برای مانیتورینگ/لانچ سبک)
3
  gradio==5.47.0
4
  gradio-client==1.13.1
5
 
6
- # HF / NLP Stack
7
  transformers==4.44.2
8
  accelerate==0.34.2
9
- datasets==2.21.0
10
- evaluate==0.4.2
11
  sentencepiece>=0.2.0
12
 
13
- # RAG / Retrieval
14
- sentence-transformers==3.0.1
15
  chromadb==0.5.5
16
  rank-bm25==0.2.2
17
- langdetect==1.0.9
18
 
19
- # ML Utils
20
- scikit-learn>=1.3.2,<1.6
21
- jsonschema>=4.22.0,<5
22
  tqdm>=4.66,<5
23
  numpy>=1.26,<3
24
-
25
- # Logging / Ops
26
- wandb>=0.17.0
27
  spaces>=0.42.0
28
 
29
- # PyTorch (Training)
30
- torch==2.3.1
31
-
32
- # Fine-tuning (LoRA/SFT/RLHF-ready)
33
- peft==0.11.1
34
- trl==0.11.4
35
-
36
- # --- Optional (NVIDIA-only; برای کوانتایز/Memory-optim) ---
37
- # bitsandbytes==0.43.1 # روی CPU/AMD نصب نکنید؛ در صورت CUDA/NVIDIA فعال کنید
38
- # xformers==0.0.26.post1 # اختیاری برای attention بهینه؛ نسخه را با CUDA سازگار تست کنید
 
1
+ # --- UI ---
 
2
  gradio==5.47.0
3
  gradio-client==1.13.1
4
 
5
+ # --- HF / NLP ---
6
  transformers==4.44.2
7
  accelerate==0.34.2
 
 
8
  sentencepiece>=0.2.0
9
 
10
+ # --- RAG (ایندکس موجود) ---
 
11
  chromadb==0.5.5
12
  rank-bm25==0.2.2
 
13
 
14
+ # --- Utils ---
 
 
15
  tqdm>=4.66,<5
16
  numpy>=1.26,<3
 
 
 
17
  spaces>=0.42.0
18
 
19
+ # توجه مهم برای ZeroGPU:
20
+ # عمداً torch اضافه نشده است؛ ZeroGPU نسخهٔ سازگار PyTorch را تزریق می‌کند.