kz209 commited on
Commit
3b35717
·
1 Parent(s): da4acea
Files changed (1) hide show
  1. utils/model.py +1 -1
utils/model.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
 
6
  class Model():
7
- def __init__(self, model="mistralai/Mistral-7B-Instruct-v0.1") -> None:
8
  pass
9
  self.tokenizer = AutoTokenizer.from_pretrained(model)
10
  self.pipeline = transformers.pipeline(
 
4
 
5
 
6
  class Model():
7
+ def __init__(self, model="tiiuae/falcon-7b-instruct") -> None:
8
  pass
9
  self.tokenizer = AutoTokenizer.from_pretrained(model)
10
  self.pipeline = transformers.pipeline(