Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

can this module download pytorch weights? #614

Open
2 tasks done
futureflsl opened this issue Aug 27, 2024 · 1 comment
Open
2 tasks done

can this module download pytorch weights? #614

futureflsl opened this issue Aug 27, 2024 · 1 comment
Labels
enhancement New feature or request

Comments

@futureflsl
Copy link

Search before asking

  • 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?

  • Yes I'd like to help by submitting a PR!
@futureflsl futureflsl added the enhancement New feature or request label Aug 27, 2024
@futureflsl futureflsl changed the title can this moduledownload pytorch weights? can this module download pytorch weights? Aug 27, 2024
@PawelPeczek-Roboflow
Copy link
Collaborator

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants