Update my_model/KBVQA.py
Browse files- 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.
|
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
|