Delta0723 commited on
Commit
2b79122
·
verified ·
1 Parent(s): 7d8b5b3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -4
requirements.txt CHANGED
@@ -1,6 +1,18 @@
1
- torch
2
- transformers
3
- peft
 
 
 
 
 
 
 
 
 
 
 
4
  accelerate
5
- bitsandbytes
6
 
 
 
 
1
+ # Núcleo de modelos
2
+ transformers>=4.34.0
3
+ torch>=2.1.0
4
+ peft>=0.7.1
5
+
6
+ # Tokenizer necesario para Mistral
7
+ sentencepiece
8
+
9
+ # Backend API
10
+ fastapi
11
+ uvicorn
12
+
13
+ # Hugging Face
14
+ huggingface_hub
15
  accelerate
 
16
 
17
+ # Utilidades adicionales
18
+ pydantic