Fix Model Parameter Init

koalpaca.py

@@ -23,6 +23,8 @@ class KoAlpaca(Model):
         self.INPUT_FORMAT = "### 질문: <INPUT>\n\n### 답변:"
         self.model.eval()
         
+        super().__init__()
+        
     def generate(self, inputs):
         inputs = self.INPUT_FORMAT.replace('<INPUT>', inputs)
         output_ids = self.model.generate(

model.py

@@ -5,10 +5,6 @@ class Model:
             placeholder:str="Input"):
         self.name = name
         self.placeholder = placeholder
-        self.model = None
-        self.tokenizer = None
-        self.gen_config = None
-        self.INPUT_FORMAT = None
         self.SPETIAL_TOKENS = ["#화자#", "#청자#", "#(남자)청자#", "#(남자)화자#", "#(여자)청자#", "(여자)화자"]
     
     def generate(self, inputs:str) -> str:

t5.py

@@ -21,6 +21,8 @@ class T5(Model):
         self.model.resize_token_embeddings(len(self.tokenizer))
         self.model.config.max_length = max_target_length
         self.tokenizer.model_max_length = max_target_length
+        
+        super().__init__()
     
     def generate(self, inputs):
         inputs = self.INPUT_FORMAT.replace("<INPUT>", inputs)