Spaces:
Runtime error
Runtime error
aiwithankit
commited on
Commit
•
ec3c939
1
Parent(s):
3208e74
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ llm = HuggingFaceLLM(
|
|
31 |
model_name="NousResearch/Llama-2-7b-hf",
|
32 |
device_map="auto",
|
33 |
# uncomment this if using CUDA to reduce memory usage
|
34 |
-
model_kwargs={"torch_dtype": torch.float16 , "load_in_8bit":False}
|
35 |
)
|
36 |
|
37 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
|
|
31 |
model_name="NousResearch/Llama-2-7b-hf",
|
32 |
device_map="auto",
|
33 |
# uncomment this if using CUDA to reduce memory usage
|
34 |
+
# model_kwargs={"torch_dtype": torch.float16 , "load_in_8bit":False}
|
35 |
)
|
36 |
|
37 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|