Rahul1872 commited on
Commit
2be06e4
Β·
verified Β·
1 Parent(s): 8288338

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -5
requirements.txt CHANGED
@@ -1,5 +1,10 @@
1
- from peft import PeftModel
2
- from transformers import AutoModelForCausalLM
3
-
4
- base_model = AutoModelForCausalLM.from_pretrained("unsloth/meta-llama-3.1-8b-instruct-unsloth-bnb-4bit")
5
- model = PeftModel.from_pretrained(base_model, "Raazi29/Nyaya-Llama-3.1-8B-Indian-Legal")
 
 
 
 
 
 
1
+ gradio>=4.0.0
2
+ transformers>=4.37.0
3
+ torch>=2.1.0
4
+ accelerate
5
+ sentencepiece
6
+ peft
7
+ bitsandbytes==0.42.0
8
+ datasets
9
+ faiss-cpu
10
+ sentence-transformers