AlirezaHSZ commited on
Commit
d32abf5
·
verified ·
1 Parent(s): 70f6911

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
 
10
  # Configuration
11
  EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
12
- LLM_MODEL_NAME = "deepseek-ai/DeepSeek-V2-Lite-Chat"
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
  # Text Splitting
 
9
 
10
  # Configuration
11
  EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
12
+ LLM_MODEL_NAME = "TheBloke/Llama-2-7B-GGUF"
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
  # Text Splitting