Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Download patch from facebook/CutLER: direct link, hf CLI and curl.
- Browser
- Download file 585 Bytes
-
https://huggingface.co/spaces/facebook/CutLER/resolve/main/patch
- Command line
-
hf download hf://spaces/facebook/CutLER/patch
-
curl -L -o patch https://huggingface.co/spaces/facebook/CutLER/resolve/main/patch
585 Bytes
| diff --git a/cutler/demo/predictor.py b/cutler/demo/predictor.py | |
| index 0920159..6fc8053 100644 | |
| --- a/cutler/demo/predictor.py | |
| +++ b/cutler/demo/predictor.py | |
| class VisualizationDemo(object): | |
| ) | |
| if "instances" in predictions: | |
| instances = predictions["instances"].to(self.cpu_device) | |
| + if self.metadata.name == '__unused': | |
| + instances.remove('pred_classes') | |
| vis_output = visualizer.draw_instance_predictions(predictions=instances) | |
| return predictions, vis_output | |