Namitg02 commited on
Commit
1791b73
·
verified ·
1 Parent(s): 4e069a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,8 +66,8 @@ from langchain.chains import create_retrieval_chain
66
  #from langchain_community.llms import Ollama
67
  #READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
68
 
69
- from transformers import AutoModel
70
- lll_model = AutoModel.from_pretrained("HuggingFaceH4/zephyr-7b-beta", load_in_4bit=True)
71
 
72
  #READER_MODEL = "HuggingFaceH4/zephyr-7b-beta"
73
 
 
66
  #from langchain_community.llms import Ollama
67
  #READER_MODEL=Ollama(model="meta-llama/Meta-Llama-Guard-2-8B")
68
 
69
+ from transformers import AutoModelForCausalLM
70
+ lll_model = AutoModelForCausalLM.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
71
 
72
  #READER_MODEL = "HuggingFaceH4/zephyr-7b-beta"
73