Spaces:

matanru
/

CapeX

Sleeping

matanru commited on Nov 26, 2024

Commit

b54a1b1

verified ·

1 Parent(s): 05b42bf

Update models/models/detectors/pam.py

Files changed (1) hide show

models/models/detectors/pam.py CHANGED Viewed

@@ -71,7 +71,7 @@ class PoseAnythingModel(BasePose):
             text_backbone_type = "bert-multilingual"
         self.text_backbone_device = "cuda" if torch.cuda.is_available() else "cpu"
-        text_backbone.to(device=self.text_backbone_device)
         if self.finetune_text_pretrained:
             if text_backbone_type == "clip":

             text_backbone_type = "bert-multilingual"
         self.text_backbone_device = "cuda" if torch.cuda.is_available() else "cpu"
+        # text_backbone.to(device=self.text_backbone_device)
         if self.finetune_text_pretrained:
             if text_backbone_type == "clip":