Remove mlperf logging (#205)

* Remove mlperf logging

Signed-off-by: Kunjan patel <kunjanp@google.com>

* Try out posting xlml test link in PR

Signed-off-by: Kunjan patel <kunjanp@google.com>

---------

Signed-off-by: Kunjan patel <kunjanp@google.com>

Author: coolkp

.github/actions/setup-miniconda/action.yml

@@ -7,7 +7,7 @@ inputs:
     description: If set to any value, dont use sudo to clean the workspace
     required: false
     type: string
-    default: "3.9"
+    default: "3.10"
   miniconda-version:
     description: Miniconda version to install
     required: false

.github/workflows/XLML.yml

@@ -0,0 +1,22 @@
+name: Add Testgrid Link to PR
+
+on:
+  pull_request:
+    types: [opened, synchronize]
+
+jobs:
+  add_testgrid_link:
+    runs-on: ubuntu-latest
+    permissions:
+      pull-requests: write
+    steps:
+      - name: Add link to PR description
+        env:
+          PR_NUMBER: ${{ github.event.pull_request.number }}
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          curl -X POST \
+            -H "Authorization: token $GITHUB_TOKEN" \
+            -H "Accept: application/vnd.github.v3+json" \
+            "https://api.github.com/repos/${{ github.repository }}/issues/$PR_NUMBER/comments" \
+            -d '{ "body": "e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid" }'

src/maxdiffusion/mllog_utils.py

@@ -21,7 +21,6 @@
 from maxdiffusion import (
     max_logging,
     pyconfig,
-    mllog_utils,
 )
 
 from maxdiffusion.train_utils import (
@@ -39,7 +38,6 @@ def train(config):
 def main(argv: Sequence[str]) -> None:
   pyconfig.initialize(argv)
   config = pyconfig.config
-  mllog_utils.train_init_start(config)
   validate_train_config(config)
   max_logging.log(f"Found {jax.device_count()} devices.")
   train(config)

src/maxdiffusion/train_flux.py

@@ -21,7 +21,6 @@
 from maxdiffusion import (
     max_logging,
     pyconfig,
-    mllog_utils,
 )
 
 from maxdiffusion.trainers.sdxl_trainer import StableDiffusionXLTrainer
@@ -39,7 +38,6 @@ def train(config):
 def main(argv: Sequence[str]) -> None:
   pyconfig.initialize(argv)
   config = pyconfig.config
-  mllog_utils.train_init_start(config)
   validate_train_config(config)
   max_logging.log(f"Found {jax.device_count()} devices.")
   train(config)