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

  Copy full SHA for 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

  Copy full SHA for e40e5dd

• small changes

  nancyhung

  committed
  454e18b

  Copy full SHA for 454e18b

• isolated changes

  nancyhung

  committed
  c8bd06f

  Copy full SHA for c8bd06f

Commits on Oct 2, 2024

• pr feedback with a print statement for testing

  nancyhung

  committed
  0d3f9ce

  Copy full SHA for 0d3f9ce

Commits on Oct 4, 2024

• some more todos and need to test

  nancyhung

  committed
  6ea8de5

  Copy full SHA for 6ea8de5

Commits on Oct 11, 2024

• need to test

  nancyhung

  committed
  81306d8

  Copy full SHA for 81306d8

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

  nancyhung

  committed
  b854bb2

  Copy full SHA for b854bb2

Commits on Oct 15, 2024

• use mlflow log model by default

  nancyhung

  committed
  bc73f65

  Copy full SHA for bc73f65

• patch push

  nancyhung

  committed
  1915042

  Copy full SHA for 1915042

Commits on Oct 22, 2024

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

  nancyhung

  committed
  bc29278

  Copy full SHA for bc29278

• add log statements

  nancyhung

  committed
  99589c7

  Copy full SHA for 99589c7

Commits on Oct 23, 2024

• add log outside of process

  nancyhung

  committed
  04ddfaa

  Copy full SHA for 04ddfaa

Commits on Oct 25, 2024

• fix

  nancyhung

  committed
  8e42217

  Copy full SHA for 8e42217

• bug

  nancyhung

  committed
  be04e3d

  Copy full SHA for be04e3d

Commits on Oct 26, 2024

• print the registered model name

  nancyhung

  committed
  5ab2cc7

  Copy full SHA for 5ab2cc7

• update the model registry prefix

  nancyhung

  committed
  79356d8

  Copy full SHA for 79356d8

• move the download code out of the if statement

  nancyhung

  committed
  4327257

  Copy full SHA for 4327257

• try registering just the model name

  nancyhung

  committed
  6c5fb05

  Copy full SHA for 6c5fb05

• connect the existing mlflow run id

  nancyhung

  committed
  bb0dd6a

  Copy full SHA for bb0dd6a

• omg it works

  nancyhung

  committed
  c5ae4ff

  Copy full SHA for c5ae4ff

Commits on Oct 29, 2024

• pr feedback

  nancyhung

  committed
  4c86e63

  Copy full SHA for 4c86e63

• add test helper

  nancyhung

  committed
  b1477bc

  Copy full SHA for b1477bc

• fix tests

  nancyhung

  committed
  e376621

  Copy full SHA for e376621

• mocking mlflow start run

  nancyhung

  committed
  e2a9d86

  Copy full SHA for e2a9d86

• fix

  nancyhung

  committed
  5784c26

  Copy full SHA for 5784c26

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

  nancyhung

  committed
  625cc29

  Copy full SHA for 625cc29

Commits on Oct 30, 2024

• pr

  nancyhung

  committed
  c939752

  Copy full SHA for c939752

• json format

  nancyhung

  committed
  9e59a21

  Copy full SHA for 9e59a21

• patches

  nancyhung

  committed
  5e13b83

  Copy full SHA for 5e13b83

• still not fully working

  nancyhung

  committed
  19862d2

  Copy full SHA for 19862d2

Commits on Oct 31, 2024

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

  nancyhung

  committed
  1eefb84

  Copy full SHA for 1eefb84

Commits on Nov 1, 2024

• overloading the config mapper still not working

  nancyhung

  committed
  9282fe0

  Copy full SHA for 9282fe0

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

  nancyhung

  committed
  9f9e027

  Copy full SHA for 9f9e027

• using irenes changes

  nancyhung

  committed
  04b520b

  Copy full SHA for 04b520b

• default name logic

  nancyhung

  committed
  687e48b

  Copy full SHA for 687e48b

• typo

  nancyhung

  committed
  65a5a1c

  Copy full SHA for 65a5a1c

• precommit

  nancyhung

  committed
  ff2f4ac

  Copy full SHA for ff2f4ac

• precommit again

  nancyhung

  committed
  67a3acc

  Copy full SHA for 67a3acc

• precommit

  nancyhung

  committed
  598d4f3

  Copy full SHA for 598d4f3

• fix tests

  nancyhung

  committed
  6c34a23

  Copy full SHA for 6c34a23

• license

  nancyhung

  committed
  a5fe322

  Copy full SHA for a5fe322

• pr ffeedback and test remove start_run

  nancyhung

  committed
  5d37fd5

  Copy full SHA for 5d37fd5

• precommit

  nancyhung

  committed
  0132611

  Copy full SHA for 0132611

• start run unnecessary

  nancyhung

  committed
  18025f7

  Copy full SHA for 18025f7

• typing

  nancyhung

  committed
  356e3b2

  Copy full SHA for 356e3b2

• fix ci

  dakinggg

  committed
  6daecb8

  Copy full SHA for 6daecb8

• fix

  dakinggg

  committed
  e3b28bf

  Copy full SHA for e3b28bf

• clean up tests

  dakinggg

  committed
  674506d

  Copy full SHA for 674506d

• fix conflict

  dakinggg

  committed
  7a423b1

  Copy full SHA for 7a423b1

• type ignore

  dakinggg

  committed
  30b6927

  Copy full SHA for 30b6927