Fix bug in get-mlperf-inference-loadgen

mlcommons · Oct 1, 2024 · 8371f0b · 8371f0b
1 parent ce56f82
commit 8371f0b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/script/app-mlperf-inference-mlcommons-python/_cm.yaml b/script/app-mlperf-inference-mlcommons-python/_cm.yaml
@@ -702,11 +702,6 @@ variations:
       torchvision:
         tags: _rocm
 
-  rocm,sdxl:
-    add_deps:
-      mlperf-implementation:
-        tags: _repo.https://github.com/gateoverflow/inference
-
   ray:
     group: framework
     add_deps_recursive:

diff --git a/script/get-mlperf-inference-loadgen/customize.py b/script/get-mlperf-inference-loadgen/customize.py
@@ -4,6 +4,7 @@
 def preprocess(i):
 
     os_info = i['os_info']
+    env = i['env']
 
     if env.get('CM_MLPERF_INFERENCE_LOADGEN_INSTALL_FROM_PIP', '') == 'yes':
         i['run_script_input']['script_name'] = "donotrun"