kz209 commited on
Commit
ebdc474
1 Parent(s): 4e6ddea
Files changed (1) hide show
  1. 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",