-
Notifications
You must be signed in to change notification settings - Fork 200
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
Stubs files for genai bindings #1158
Conversation
def __init__(self, models_path: str, scheduler_config: SchedulerConfig, device: str, properties: dict[str, typing.Any] = {}, tokenizer_properties: dict[str, typing.Any] = {}) -> None: | ||
... | ||
@typing.overload | ||
def __init__(self, models_path: str, tokenizer: Tokenizer, scheduler_config: SchedulerConfig, device: str, properties: dict[str, typing.Any] = {}) -> None: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
properties
are not aligned with other pipelines where we have kwargs
.
Maybe we can drop tokenizer_properties
(if they are not used anywhere) and use kwargs
only
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think tokenizer_properties are needed for OVMS to set throughput hint
from openvino_genai.py_openvino_genai import draft_model | ||
import os as os | ||
from . import py_openvino_genai | ||
__all__ = ['Adapter', 'AdapterConfig', 'AggregationMode', 'AutoencoderKL', 'CLIPTextModel', 'CLIPTextModelWithProjection', 'CacheEvictionConfig', 'ChunkStreamerBase', 'ContinuousBatchingPipeline', 'CppStdGenerator', 'DecodedResults', 'EncodedResults', 'GenerationConfig', 'GenerationResult', 'Generator', 'ImageGenerationConfig', 'LLMPipeline', 'PerfMetrics', 'RawPerfMetrics', 'Scheduler', 'SchedulerConfig', 'StopCriteria', 'StreamerBase', 'Text2ImagePipeline', 'TokenizedInputs', 'Tokenizer', 'UNet2DConditionModel', 'VLMPipeline', 'WhisperGenerationConfig', 'WhisperPipeline', 'draft_model', 'openvino', 'os', 'py_openvino_genai'] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should we avoid providing 'os'
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks like it is bybind11-stubgen bug, mypy-stubgen doesn't export os.
Commited mypy-stubgen output for comparison: 1234257
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks like it's done because of os.PathLike
In general, it's not a problem as os
is standard module. I think we can merge pybind11-stubgen
output as start point.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's commit pybind11-stubgen
output
Done. |
- #1158 - #1178 - #1214 - #1243 - #1253 - #1259 - #1266 - #1271 - #1278 - #1280 - #1284 - e4a86f6 - #1246 - #958 --------- Co-authored-by: Anastasiia Pnevskaia <[email protected]> Co-authored-by: Helena Kloosterman <[email protected]> Co-authored-by: Vladimir Zlobin <[email protected]> Co-authored-by: Dmitry Matveev <[email protected]> Co-authored-by: Anna Likholat <[email protected]> Co-authored-by: Alina Kladieva <[email protected]>
CVS-156607