tdecae commited on
Commit
863bb17
·
verified ·
1 Parent(s): 53a7eb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,7 +42,8 @@ vectorstore.add_texts(texts=texts, metadatas=metadatas, embeddings=embeddings)
42
  vectorstore.persist()
43
 
44
 
45
- model_name = "deepseek-ai/deepseek-llm-7b-instruct"
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_name)
47
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
48
 
 
42
  vectorstore.persist()
43
 
44
 
45
+ # model_name = "deepseek-ai/deepseek-llm-7b-instruct"
46
+ model_name = "google/flan-t5-large"
47
  tokenizer = AutoTokenizer.from_pretrained(model_name)
48
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
49