Code

Add MLflow log_model option#1544

Code

Merged

dakinggg merged 51 commits intomainmosaicml/llm-foundry:mainfrom nancy/log-modelmosaicml/llm-foundry:nancy/log-modelCopy head branch name to clipboardNov 1, 2024

+282-121

Commits

Commits on Sep 24, 2024

• Register model with MLflow PySDK now that retries are baked in. This cleans up the code a little and prevents us from having forked logic in Composer to fetch by run_id

  nancyhung

  committed
  06d77db

• Register model with MLflow PySDK now that retries are baked in. This cleans up the code a little and prevents us from having forked logic in Composer to fetch by run_id

  nancyhung

  committed
  e40e5dd

• small changes

  nancyhung

  committed
  454e18b

• isolated changes

  nancyhung

  committed
  c8bd06f

Commits on Oct 2, 2024

• pr feedback with a print statement for testing

  nancyhung

  committed
  0d3f9ce

Commits on Oct 4, 2024

• some more todos and need to test

  nancyhung

  committed
  6ea8de5

Commits on Oct 11, 2024

• need to test

  nancyhung

  committed
  81306d8

• Merge branch 'main' into nancy/log-model

  nancyhung

  committed
  b854bb2

Commits on Oct 15, 2024

• use mlflow log model by default

  nancyhung

  committed
  bc73f65

• patch push

  nancyhung

  committed
  1915042

Commits on Oct 22, 2024

• Merge branch 'main' into nancy/log-model

  nancyhung

  committed
  bc29278

• add log statements

  nancyhung

  committed
  99589c7

Commits on Oct 23, 2024

• add log outside of process

  nancyhung

  committed
  04ddfaa

Commits on Oct 25, 2024

• fix

  nancyhung

  committed
  8e42217

• bug

  nancyhung

  committed
  be04e3d

Commits on Oct 26, 2024

• print the registered model name

  nancyhung

  committed
  5ab2cc7

• update the model registry prefix

  nancyhung

  committed
  79356d8

• move the download code out of the if statement

  nancyhung

  committed
  4327257

• try registering just the model name

  nancyhung

  committed
  6c5fb05

• connect the existing mlflow run id

  nancyhung

  committed
  bb0dd6a

• omg it works

  nancyhung

  committed
  c5ae4ff

Commits on Oct 29, 2024

• pr feedback

  nancyhung

  committed
  4c86e63

• add test helper

  nancyhung

  committed
  b1477bc

• fix tests

  nancyhung

  committed
  e376621

• mocking mlflow start run

  nancyhung

  committed
  e2a9d86

• fix

  nancyhung

  committed
  5784c26

• Merge branch 'main' into nancy/log-model

  nancyhung

  committed
  625cc29

Commits on Oct 30, 2024

• pr

  nancyhung

  committed
  c939752

• json format

  nancyhung

  committed
  9e59a21

• patches

  nancyhung

  committed
  5e13b83

• still not fully working

  nancyhung

  committed
  19862d2

Commits on Oct 31, 2024

• fixed the final_register_only test case. now need to pass the others

  nancyhung

  committed
  1eefb84

Commits on Nov 1, 2024

• overloading the config mapper still not working

  nancyhung

  committed
  9282fe0

• Merge branch 'main' into nancy/log-model

  nancyhung

  committed
  9f9e027

• using irenes changes

  nancyhung

  committed
  04b520b

• default name logic

  nancyhung

  committed
  687e48b

• typo

  nancyhung

  committed
  65a5a1c

• precommit

  nancyhung

  committed
  ff2f4ac

• precommit again

  nancyhung

  committed
  67a3acc

• precommit

  nancyhung

  committed
  598d4f3

• fix tests

  nancyhung

  committed
  6c34a23

• license

  nancyhung

  committed
  a5fe322

• pr ffeedback and test remove start_run

  nancyhung

  committed
  5d37fd5

• precommit

  nancyhung

  committed
  0132611

• start run unnecessary

  nancyhung

  committed
  18025f7

• typing

  nancyhung

  committed
  356e3b2

• fix ci

  dakinggg

  committed
  6daecb8

• fix

  dakinggg

  committed
  e3b28bf

• clean up tests

  dakinggg

  committed
  674506d

• fix conflict

  dakinggg

  committed
  7a423b1

• type ignore

  dakinggg

  committed
  30b6927