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.

Sign up for GitHub

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

Jump to bottom

Stubs files for genai bindings #1158

Draft
wants to merge 4 commits into
base: master
Choose a base branch
from
+1,811 −83
Draft

Stubs files for genai bindings #1158

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
44ddee6
Fixed warnings from pybind11-stubgen.
popovaan Nov 5, 2024
699859f
Fixws wrong parameters names.
popovaan Nov 6, 2024
4e3446e
Generated pyi files.
popovaan Nov 6, 2024
43f5b97
Merge branch 'master' into pyi_file
popovaan Nov 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 38 additions & 0 deletions src/python/openvino_genai/__init__.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
"""
openvino genai module namespace, exposing pipelines and configs to create these pipelines.
"""
from __future__ import annotations
import openvino as openvino
from openvino_genai.py_openvino_genai import Adapter
from openvino_genai.py_openvino_genai import AdapterConfig
from openvino_genai.py_openvino_genai import AggregationMode
from openvino_genai.py_openvino_genai import AutoencoderKL
from openvino_genai.py_openvino_genai import CLIPTextModel
from openvino_genai.py_openvino_genai import CLIPTextModelWithProjection
from openvino_genai.py_openvino_genai import CacheEvictionConfig
from openvino_genai.py_openvino_genai import ContinuousBatchingPipeline
from openvino_genai.py_openvino_genai import CppStdGenerator
from openvino_genai.py_openvino_genai import DecodedResults
from openvino_genai.py_openvino_genai import EncodedResults
from openvino_genai.py_openvino_genai import GenerationConfig
from openvino_genai.py_openvino_genai import GenerationResult
from openvino_genai.py_openvino_genai import Generator
from openvino_genai.py_openvino_genai import LLMPipeline
from openvino_genai.py_openvino_genai import PerfMetrics
from openvino_genai.py_openvino_genai import RawPerfMetrics
from openvino_genai.py_openvino_genai import Scheduler
from openvino_genai.py_openvino_genai import SchedulerConfig
from openvino_genai.py_openvino_genai import StopCriteria
from openvino_genai.py_openvino_genai import StreamerBase
from openvino_genai.py_openvino_genai import Text2ImagePipeline
from openvino_genai.py_openvino_genai import TokenizedInputs
from openvino_genai.py_openvino_genai import Tokenizer
from openvino_genai.py_openvino_genai import UNet2DConditionModel
from openvino_genai.py_openvino_genai import VLMPipeline
from openvino_genai.py_openvino_genai import WhisperGenerationConfig
from openvino_genai.py_openvino_genai import WhisperPipeline
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', 'ContinuousBatchingPipeline', 'CppStdGenerator', 'DecodedResults', 'EncodedResults', 'GenerationConfig', 'GenerationResult', 'Generator', 'LLMPipeline', 'PerfMetrics', 'RawPerfMetrics', 'Scheduler', 'SchedulerConfig', 'StopCriteria', 'StreamerBase', 'Text2ImagePipeline', 'TokenizedInputs', 'Tokenizer', 'UNet2DConditionModel', 'VLMPipeline', 'WhisperGenerationConfig', 'WhisperPipeline', 'draft_model', 'openvino', 'os', 'py_openvino_genai']
__version__: str = '2024.5.0.0'
Loading
Loading