Preview Flyte 2 for productionHosted on Union.ai
Hugging Face

Hugging Face

flytekitplugins-huggingface

FlytekitData & DataFramehuggingfacedatasetsnlpml

Hugging Face is a community and data science platform that provides: Tools that enable users to build, train and deploy ML models based on open source (OS) code and technologies

Install
pip install flytekitplugins-huggingface

Quick Start(example, may need adjustment)

pip install flytekitplugins-huggingface

from flytekit import task, workflow
from flytekitplugins.huggingface import HuggingFaceDatasetToParquetEncodingHandler, ParquetToHuggingFaceDatasetDecodingHandler

config = HuggingFaceDatasetToParquetEncodingHandler(...)

@task
def my_task() -> None:
    ...

@workflow
def my_workflow() -> None:
    my_task()

Available Imports (2)

handlerHuggingFaceDatasetToParquetEncodingHandler

Configuration type for Hugging Face.

from flytekitplugins.huggingface import HuggingFaceDatasetToParquetEncodingHandler

handlerParquetToHuggingFaceDatasetDecodingHandler

Configuration type for Hugging Face.

from flytekitplugins.huggingface import ParquetToHuggingFaceDatasetDecodingHandler

Dependencies

datasets

Related Plugins

Package Info

Min Flytekit1.3.0
Modules2

Downloads

Last day48
Last week229
Last month1,413