m7mdal7aj commited on
Commit
750806d
1 Parent(s): 4d7e92d

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. my_model/KBVQA.py +1 -1
my_model/KBVQA.py CHANGED
@@ -49,7 +49,7 @@ class KBVQA():
49
 
50
  def __init__(self):
51
 
52
- self.model_name: str = config.KBVQA_MODEL_NAME
53
  self.quantization: str = config.QUANTIZATION
54
  self.max_context_window: int = config.MAX_CONTEXT_WINDOW
55
  self.add_eos_token: bool = config.ADD_EOS_TOKEN
 
49
 
50
  def __init__(self):
51
 
52
+ self.kbvqa_model_name: str = config.KBVQA_MODEL_NAME
53
  self.quantization: str = config.QUANTIZATION
54
  self.max_context_window: int = config.MAX_CONTEXT_WINDOW
55
  self.add_eos_token: bool = config.ADD_EOS_TOKEN