Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -294,10 +294,14 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
     """
   
     if force_reload:
-        loading_message = 'Force Reloading model.. this should take no more than a few minutes!'
+        free_gpu_resources()
+        loading_message = 'Reloading model.. this should take no more than 2 or 3 minutes!'
         try:
             del kbvqa
+            free_gpu_resources()
+            free_gpu_resources()
         except:
+            free_gpu_resources()
             free_gpu_resources()
             pass
         free_gpu_resources()