Spaces:
Sleeping
Sleeping
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
|