Thanks to visit codestin.com
Credit goes to github.com

Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

[TextGeneration] Add helper function to parse model path from args#1583

Merged
dsikka merged 5 commits into
mainfrom
update_create
Feb 7, 2024
Merged

[TextGeneration] Add helper function to parse model path from args#1583
dsikka merged 5 commits into
mainfrom
update_create

Conversation

@dsikka

@dsikka dsikka commented Feb 5, 2024

Copy link
Copy Markdown
Contributor

Summary

The following now works:

from deepsparse import TextGeneration
model_path = "hf:mgoin/TinyStories-1M-ds"

pipe = TextGeneration(model_path)

This still works:

model_path  = TextGeneration(model_path=model_path)

bfineran
bfineran previously approved these changes Feb 5, 2024
Comment thread src/deepsparse/pipeline.py Outdated
Comment thread src/deepsparse/pipeline.py
@dsikka dsikka requested review from bfineran and rahul-tuli February 6, 2024 04:33
@dsikka dsikka merged commit c54461a into main Feb 7, 2024
@dsikka dsikka deleted the update_create branch February 7, 2024 18:08
dsikka added a commit that referenced this pull request Feb 7, 2024
…1583)

* add helper function to parse model path from args

* update model path

* revert cli changes

* remove empty args
bfineran pushed a commit that referenced this pull request Feb 7, 2024
…1583)

* add helper function to parse model path from args

* update model path

* revert cli changes

* remove empty args
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants