Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
·
3b35717
1
Parent(s):
da4acea
update
Browse files- 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="
|
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(
|