This pull request introduces a new feature that allows for fine-tuning of transformer models from HuggingFace within our LLM (Large Language Models) gateway. It includes backend implementations for initiating fine-tuning and checking the status of these operations, as well as frontend adjustments to enable this functionality within our studio app.
Summary
Added backend API endpoints in gateway/internal/api/v1/finetune.go for:
Initiating fine-tuning of models with HuggingFace (InitiateFineTuning)
Checking the status of fine-tuning operations (CheckFineTuningStatus)
Implemented a new provider in gateway/internal/provider/huggingface/huggingface.go to interact with HuggingFace's API for model fine-tuning.
Updated the frontend component modelselector.tsx in playgrounds/apps/studio/app/(llm)/playground/components to include a toggle for selecting models specifically for fine-tuning.
Modified the useModelFetch.tsx hook in playgrounds/apps/studio/app/(llm)/playground/hooks to support fetching models based on whether fine-tuning is enabled or not.
These changes collectively introduce the capability to fine-tune transformer models from HuggingFace, enhancing the flexibility and power of MissingStudios LLM offerings.
š Unity AI Development Trusts Sweep For Quick Development!
Description
This pull request introduces a new feature that allows for fine-tuning of transformer models from HuggingFace within our LLM (Large Language Models) gateway. It includes backend implementations for initiating fine-tuning and checking the status of these operations, as well as frontend adjustments to enable this functionality within our studio app.
Summary
gateway/internal/api/v1/finetune.go
for:InitiateFineTuning
)CheckFineTuningStatus
)gateway/internal/provider/huggingface/huggingface.go
to interact with HuggingFace's API for model fine-tuning.modelselector.tsx
inplaygrounds/apps/studio/app/(llm)/playground/components
to include a toggle for selecting models specifically for fine-tuning.useModelFetch.tsx
hook inplaygrounds/apps/studio/app/(llm)/playground/hooks
to support fetching models based on whether fine-tuning is enabled or not.These changes collectively introduce the capability to fine-tune transformer models from HuggingFace, enhancing the flexibility and power of MissingStudios LLM offerings.
š Unity AI Development Trusts Sweep For Quick Development!