roboflow / inference

A fast, easy-to-use, production-ready inference server for computer vision supporting deployment of many popular model architectures and fine-tuned models.

https://inference.roboflow.com

Other

1.36k stars 126 forks source link

can this module download pytorch weights? #614

Open futureflsl opened 2 months ago

futureflsl commented 2 months ago

Search before asking

[X] I have searched the Inference issues and found no similar feature requests.

Description

from inference import get_model

load a pre-trained yolov8n model

model = get_model(model_id="",api_key='')

this function can download onnx weights,but I just want pytorch weights,can this function download pytorch weights?

Use case

No response

Additional

No response

Are you willing to submit a PR?

[X] Yes I'd like to help by submitting a PR!

PawelPeczek-Roboflow commented 2 months ago

Hi there

get_model(...) is supposed to pull the model from Roboflow platform, we do ship models via onnx now, probably you cannot run arbitrary torch model