Spaces:

XAI
/

PEEB

Sleeping

Peijie commited on Apr 26

Commit

c8636e8

•

1 Parent(s): 401a164

load model in cpu first

Files changed (1) hide show

utils/load_model.py CHANGED Viewed

@@ -13,7 +13,6 @@ from transformers import OwlViTProcessor, OwlViTForObjectDetection
 from .model import OwlViTForClassification
-@gpu_decorator
 def load_xclip(device: str = "cuda:0",
                n_classes: int = 183,
                use_teacher_logits: bool = False,

 from .model import OwlViTForClassification
 def load_xclip(device: str = "cuda:0",
                n_classes: int = 183,
                use_teacher_logits: bool = False,