debug logits

interfaces/cap.py

@@ -105,7 +105,9 @@ def predict(text, model_id, tokenizer_id):
     inputs = {k: v.to(device) for k, v in inputs.items()}
 
     with torch.no_grad():
-        logits = model(inputs["input_ids"], inputs["attention_mask"]).logits
+        output = model(inputs["input_ids"], inputs["attention_mask"])
+        print(output)
+        logits = output["logits"]
         
     release_model(model, model_id)
 

utils.py

@@ -57,6 +57,7 @@ for domain in domains_illframes.values():
 tokenizers = ["xlm-roberta-large"]
 
 def download_hf_models():
+    return ### DEBUG
     # Ensure the JIT model directory exists
     os.makedirs(JIT_DIR, exist_ok=True)