Spaces:

yonigozlan
/

RTDETR-compiled-speed-test

Runtime error

yonigozlan HF Staff commited on Sep 17, 2024

Commit

40a6300

1 Parent(s): da01a2d

enable 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=True,
     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=False,
     torch_dtype=torch.float16,
 ).to(device)
 model_compiled = torch.compile(