aiwithankit commited on
Commit
ec3c939
1 Parent(s): 3208e74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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