alexkueck commited on
Commit
acb5706
·
1 Parent(s): d063f92

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -68,8 +68,8 @@ def generate_prompt_with_history(text, history, tokenizer, max_length=2048):
68
 
69
  #tokenizer = AutoTokenizer.from_pretrained("project-baize/baize-v2-7b")
70
  #model = AutoModelForCausalLM.from_pretrained("project-baize/baize-v2-7b")
71
- tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
72
- model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
73
 
74
 
75
  def load_tokenizer_and_model(base_model,load_8bit=False):
 
68
 
69
  #tokenizer = AutoTokenizer.from_pretrained("project-baize/baize-v2-7b")
70
  #model = AutoModelForCausalLM.from_pretrained("project-baize/baize-v2-7b")
71
+ #tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
72
+ #model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
73
 
74
 
75
  def load_tokenizer_and_model(base_model,load_8bit=False):