Add local agent #23438
Merged
Add local agent #23438
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
The documentation is not available anymore as the PR was closed or merged. |
LysandreJik
approved these changes
May 18, 2023
Comment on lines
+561
to
+562
| model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path, **kwargs) | ||
| tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path, **kwargs) |
gante
reviewed
May 18, 2023
| return result | ||
|
|
||
|
|
||
| class StopSequenceCriteria(StoppingCriteria): |
There was a problem hiding this comment.
Can we:
- move it to the stopping criteria file? (I've seen a user asking for this)
- generalize it for the batched input case? (or add a todo, I'd be happy to expand it :D )
There was a problem hiding this comment.
Maybe you can move it as you add support for the batched case? I don't need it for the agents and it's not an obvious thing to do (that's the reason I didn't put this in the stopping criteria file by the way).
There was a problem hiding this comment.
Sure, I can take care of it afterwards 👍
sheonhan
pushed a commit
to sheonhan/transformers
that referenced
this pull request
Jun 1, 2023
* Add local agent * Document LocalAgent
gojiteji
pushed a commit
to gojiteji/transformers
that referenced
this pull request
Jun 5, 2023
* Add local agent * Document LocalAgent
novice03
pushed a commit
to novice03/transformers
that referenced
this pull request
Jun 23, 2023
* Add local agent * Document LocalAgent
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What does this PR do?
This PR adds support for a local agent running the model to generate code on the user's machine.