Spaces:
Sleeping
Sleeping
update
Browse files- models/vsa_model.py +1 -0
models/vsa_model.py
CHANGED
@@ -278,6 +278,7 @@ class VisionSearchAssistant:
|
|
278 |
the $\mathcal{F}_{vlm}(cdot)$ in the paper.
|
279 |
|
280 |
"""
|
|
|
281 |
def __init__(
|
282 |
self,
|
283 |
search_model: str = "internlm/internlm2_5-7b-chat",
|
|
|
278 |
the $\mathcal{F}_{vlm}(cdot)$ in the paper.
|
279 |
|
280 |
"""
|
281 |
+
@spaces.GPU
|
282 |
def __init__(
|
283 |
self,
|
284 |
search_model: str = "internlm/internlm2_5-7b-chat",
|