Spaces:
Sleeping
Sleeping
Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +1 -1
my_model/KBVQA.py
CHANGED
@@ -4,7 +4,7 @@ import copy
|
|
4 |
import os
|
5 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
6 |
from typing import Optional
|
7 |
-
from my_model.
|
8 |
from my_model.captioner.image_captioning import ImageCaptioningModel
|
9 |
from my_model.object_detection import ObjectDetector
|
10 |
|
|
|
4 |
import os
|
5 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
6 |
from typing import Optional
|
7 |
+
from my_model.gen_utilities import free_gpu_resources
|
8 |
from my_model.captioner.image_captioning import ImageCaptioningModel
|
9 |
from my_model.object_detection import ObjectDetector
|
10 |
|