Hugging Face's logo

Spaces:
XAI
/
PEEB
Running on Zero

App Files Community
1
Peijie commited on
Commit
c8636e8
·
1 Parent(s): 401a164

load model in cpu first

Browse files
Files changed (1) hide show
  1. utils/load_model.py +0 -1
utils/load_model.py CHANGED
@@ -13,7 +13,6 @@ from transformers import OwlViTProcessor, OwlViTForObjectDetection
13
 
14
  from .model import OwlViTForClassification
15
 
16
- @gpu_decorator
17
  def load_xclip(device: str = "cuda:0",
18
  n_classes: int = 183,
19
  use_teacher_logits: bool = False,
 
13
 
14
  from .model import OwlViTForClassification
15
 
 
16
  def load_xclip(device: str = "cuda:0",
17
  n_classes: int = 183,
18
  use_teacher_logits: bool = False,