Ashim Mahara

Results 16 comments of Ashim Mahara

@fxmarty here is the PR: https://github.com/huggingface/optimum/pull/931

@fxmarty is there any way to make the pretrained deformable-detr models compatible with the new code? I tried exporting `SenseTime/deformable-detr` after changing the `disable_custom_kernels` to `True` but it still throws...

@iliesaya did you move the `img` to `mps`? I faced the same issue when I didn't move my input to the GPU.

``` for img in dataset: img = img.to('mps') ## or img.to(device) since you have already set device to mps in the previous line this_res = inference_detector(model, img) ``` If you...

I don't know what you're doing inside `inference_detector()` but if it's like ``` if next(model.parameters()).is_cuda: # scatter to specified GPU data = scatter(data, [device])[0] else: for m in model.modules(): assert...