Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
ebdc474
1
Parent(s):
4e6ddea
update
Browse files- utils/model.py +0 -1
utils/model.py
CHANGED
@@ -5,7 +5,6 @@ import torch
|
|
5 |
|
6 |
class Model():
|
7 |
def __init__(self, model="lmsys/vicuna-7b-v1.5") -> None:
|
8 |
-
pass
|
9 |
self.tokenizer = AutoTokenizer.from_pretrained(model)
|
10 |
self.pipeline = transformers.pipeline(
|
11 |
"text-generation",
|
|
|
5 |
|
6 |
class Model():
|
7 |
def __init__(self, model="lmsys/vicuna-7b-v1.5") -> None:
|
|
|
8 |
self.tokenizer = AutoTokenizer.from_pretrained(model)
|
9 |
self.pipeline = transformers.pipeline(
|
10 |
"text-generation",
|