cnzzx commited on
Commit
686a863
1 Parent(s): a014dd2
Files changed (1) hide show
  1. 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",