Hugging Face's logo

Spaces:
ohalkhateeb
/
Test3
Sleeping

App Files Community
ohalkhateeb commited on
Commit
db7c51f
·
verified ·
1 Parent(s): 10afeb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ embedding_model = SentenceTransformer("sentence-transformers/paraphrase-multilin
30
 
31
  # Load Jais model and tokenizer
32
  model_name = "inceptionai/jais-13b"
33
- tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
34
  model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN)
35
 
36
  # RAG function
 
30
 
31
  # Load Jais model and tokenizer
32
  model_name = "inceptionai/jais-13b"
33
+ tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
34
  model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN)
35
 
36
  # RAG function