Andrei Vishniakov

Fix `log_model_metadata` with explicit name and version

2

## Describe changes Before the fix `log_model_metadata` failed if called inside a step with provided `model_name` and `model_version` but without a step context. The same applies to the fact that...

Huggingface ModelRegistry

2

## Describe changes I implemented ModelRegistry component to work with Huggingface repos. This PR is a draft for early feedback collection. **Adding looms with show case**: - Part 1: https://www.loom.com/share/bf4401e3969d48c686d07492b9b6b10c...

Multi GPU with PEFT on LLM

6

This PR brings the multi-GPU DDP showcase to the PEFT training. There are some routine steps, which can be automated in ZenML core. We will create follow-up tickets for that...

LLM Finetune with PEFT

4

This PR brings in the pipeline to run Mistral model fine-tuning using PEFT library on Viggio dataset. Key highlights: - Project has one pipeline to run the full cycle of...

Add `id_` as part of the Model config

1

## Describe changes I added `id_` as a public field of the Model. It can be used to configure a Model Version using ID or will be used to represent...

Accelerate scaler

6

## Describe changes This PR introduces the concept of Scalers, which can be passed into the steps' definitions to allow step run on some parallelization engine. Scalers included here: -...

Self-cleaning integration tests

3

## Describe changes I implemented logic to clean-up after all tests we run in `tests/integration` folder and removed `clean_client` where it was used, so tests are running really on REST...

[WIP] Stack wizard CLI + Endpoints

2

## Describe changes WIP ## Pre-requisites Please ensure you have done the following: - [ ] I have read the **CONTRIBUTING.md** document. - [ ] If my change requires a...

TF tests + various integration (un)install improvements

1

## Describe changes Not an actual fix for 3.8 and TF, but: - I allowed all runtimes for `tests/integration/examples/test_tensorflow.py` except Ubuntu/3.8 only (before was banned for 3.8,3.10 and 3.11 on...

[WIP] Python 3.12

3

## Describe changes I implemented/fixed _ to achieve _. ## Pre-requisites Please ensure you have done the following: - [ ] I have read the **CONTRIBUTING.md** document. - [ ]...