ndn1954 commited on
Commit
add5f12
1 Parent(s): a67dec5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ import streamlit as st
23
  #from transformers import AutoModel
24
  #model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-Chat-GGML")
25
 
26
- from transformers import AutoModelForCausalLM
27
  model_id="TheBloke/Llama-2-7B-Chat-GGML"
28
  tokenizer=AutoTokenizer.from_pretrained(model_id)
29
  model=AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
 
23
  #from transformers import AutoModel
24
  #model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-Chat-GGML")
25
 
26
+ from transformers import AutoModelForCausalLM, AutoTokenizer
27
  model_id="TheBloke/Llama-2-7B-Chat-GGML"
28
  tokenizer=AutoTokenizer.from_pretrained(model_id)
29
  model=AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)