Spaces:

yonigozlan
/

RTDETR-compiled-speed-test

Sleeping

yonigozlan HF staff commited on Sep 17

Commit

da01a2d

•

1 Parent(s): 2037a75

disable custom kernel

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 processor = AutoProcessor.from_pretrained("PekingU/rtdetr_r50vd_coco_o365")
 model = AutoModelForObjectDetection.from_pretrained(
     "PekingU/rtdetr_r50vd_coco_o365",
-    disable_custom_kernels=False,
     torch_dtype=torch.float16,
 ).to(device)
 model_compiled = torch.compile(

 processor = AutoProcessor.from_pretrained("PekingU/rtdetr_r50vd_coco_o365")
 model = AutoModelForObjectDetection.from_pretrained(
     "PekingU/rtdetr_r50vd_coco_o365",
+    disable_custom_kernels=True,
     torch_dtype=torch.float16,
 ).to(device)
 model_compiled = torch.compile(