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

Openai triton server #104

Draft
wants to merge 66 commits into
base: main
Choose a base branch
from
Draft

Openai triton server #104

wants to merge 66 commits into from

Conversation

nnshah1
Copy link
Contributor

@nnshah1 nnshah1 commented Aug 2, 2024

No description provided.

if "text_output" in response.outputs:
try:
return response.outputs["text_output"].to_string_array()[0]
except:

Check notice

Code scanning / CodeQL

Except block handles 'BaseException' Note

Except block directly handles BaseException.
import argparse
import time
import uuid
from typing import Optional, Union

Check notice

Code scanning / CodeQL

Unused import Note

Import of 'Optional' is not used.
Model,
ObjectType,
)
from transformers import AutoTokenizer, PreTrainedTokenizer, PreTrainedTokenizerFast

Check notice

Code scanning / CodeQL

Unused import Note

Import of 'AutoTokenizer' is not used.


class FunctionParameters(BaseModel):
pass

Check warning

Code scanning / CodeQL

Unnecessary pass Warning

Unnecessary 'pass' statement.
Comment on lines +112 to +121
# if "BaichuanTokenizer" in str(e):
# # This is for the error "'BaichuanTokenizer' object has no
# # attribute 'sp_model'".
# tokenizer = BaichuanTokenizer.from_pretrained(
# tokenizer_name,
# *args,
# trust_remote_code=trust_remote_code,
# tokenizer_revision=tokenizer_revision,
# **kwargs)
# else:

Check notice

Code scanning / CodeQL

Commented-out code Note

This comment appears to contain commented-out code.
@dongs0104
Copy link

hello i have question about triton server running using python code import tritonserver

i have 4 gpus for serve, so i have to set world size 4 but i can not found any option or tutorials, if you can help I'd be so grateful.

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

Successfully merging this pull request may close these issues.

3 participants