Fixes some WIP models in the docs

GATEOverflow · Jun 5, 2024 · 5e49864 · 5e49864
1 parent 63fcc60
commit 5e49864
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -24,15 +24,21 @@ def mlperf_inference_implementation_readme(spaces, model, implementation):
                  frameworks = [ "Pytorch" ]
 
         elif implementation == "nvidia":
+            if model in [ "sdxl", "llama2-99", "llama2-99.9" ]:
+                 return pre_space+"    WIP"
             devices = [ "CUDA" ]
             frameworks = [ "TensorRT" ]
+
         elif implementation == "intel":
             if model not in [ "bert-99", "bert-99.9", "gptj-99", "gptj-99.9" ]:
                  return pre_space+"    WIP"
             devices = [ "CPU" ]
             frameworks = [ "Pytorch" ]
 
         elif implementation == "qualcomm":
+            if model not in [ "resnet50", "retinanet", "bert-99", "bert-99.9" ]:
+                 return pre_space+"    WIP"
+
             devices = [ "QAIC" ]
             frameworks = [ "Glow" ]