Apply patch to model lists.

models/turbine_models/custom_models/torchbench/timm_models.json

@@ -1,14 +0,0 @@
-{
-    "timm_efficientnet": {
-        "dim": 128
-    },
-    "timm_regnet": {
-        "dim": 128
-    },
-    "timm_resnest": {
-        "dim": 256
-    },
-    "timm_vovnet": {
-        "dim": 128
-    }
-}

models/turbine_models/custom_models/torchbench/torchbench_models.json

@@ -1,5 +1,50 @@
 {
+    "LearningToPaint": {
+        "dim": 1024
+    },
+    "alexnet": {
+        "dim": 1024
+    },
+    "densenet121": {
+        "dim": 64
+    },
+    "mnasnet1_0": {
+        "dim": 256
+    },
+    "mobilenet_v2": {
+        "dim": 128
+    },
+    "mobilenet_v3_large": {
+        "dim": 256
+    },
     "pytorch_unet": {
         "dim": 8
+    },
+    "resnet18": {
+        "dim": 512
+    },
+    "resnet50": {
+        "dim": 128
+    },
+    "resnext50_32x4d": {
+        "dim": 128
+    },
+    "shufflenet_v2_x1_0": {
+        "dim": 512
+    },
+    "squeezenet1_1": {
+        "dim": 512
+    },
+    "timm_efficientnet": {
+        "dim": 128
+    },
+    "timm_regnet": {
+        "dim": 128
+    },
+    "timm_resnest": {
+        "dim": 256
+    },
+    "timm_vovnet": {
+        "dim": 128
     }
 }