Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +2 -1
my_model/KBVQA.py
CHANGED
|
@@ -185,7 +185,8 @@ class KBVQA:
|
|
| 185 |
free_gpu_resources()
|
| 186 |
|
| 187 |
free_gpu_resources()
|
| 188 |
-
|
|
|
|
| 189 |
|
| 190 |
|
| 191 |
def format_prompt(self, current_query: str, history: Optional[str] = None, sys_prompt: Optional[str] = None, caption: str = None, objects: Optional[str] = None) -> str:
|
|
|
|
| 185 |
free_gpu_resources()
|
| 186 |
|
| 187 |
free_gpu_resources()
|
| 188 |
+
|
| 189 |
+
return prepare_kbvqa_model(only_reload_detection_model=False, force_reload=True)
|
| 190 |
|
| 191 |
|
| 192 |
def format_prompt(self, current_query: str, history: Optional[str] = None, sys_prompt: Optional[str] = None, caption: str = None, objects: Optional[str] = None) -> str:
|