update

utils/model.py

@@ -2,7 +2,7 @@ from transformers import AutoTokenizer
 import transformers
 import torch
 
-class Model():
+class Model(torch.nn.Module):
     number_of_models = 0
     __model_list__ = [
         "lmsys/vicuna-7b-v1.5",
@@ -10,6 +10,8 @@ class Model():
     ]
 
     def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
+        super(Model, self).__init__()
+        
         self.tokenizer = AutoTokenizer.from_pretrained(model_name)
         self.name = model_name
         self.pipeline = transformers.pipeline(