diff --git a/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py b/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py
index 25965d8103..f3ee0b587d 100644
--- a/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py
+++ b/cm-mlops/script/app-image-classification-torch-py/src/pytorch_classify_preprocessed.py
@@ -5,6 +5,8 @@
 import os
 import shutil
 import numpy as np
+
+
 import torch
 import torchvision.models as models
 
diff --git a/cm-mlops/script/get-generic-python-lib/_cm.json b/cm-mlops/script/get-generic-python-lib/_cm.json
index 8a4575ca33..d59095ba06 100644
--- a/cm-mlops/script/get-generic-python-lib/_cm.json
+++ b/cm-mlops/script/get-generic-python-lib/_cm.json
@@ -59,7 +59,6 @@
     "jax_cuda": {
       "env": {
         "CM_GENERIC_PYTHON_PACKAGE_NAME": "jax[cuda]",
-        "CM_GENERIC_PYTHON_PIP_EXTRA": "-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html",
         "CM_JAX_VERSION_EXTRA":"CUDA"
       },
       "new_env_keys": [
diff --git a/cm-mlops/script/install-generic-python-lib/_cm.json b/cm-mlops/script/install-generic-python-lib/_cm.json
index 61685bdfec..716a9442f3 100644
--- a/cm-mlops/script/install-generic-python-lib/_cm.json
+++ b/cm-mlops/script/install-generic-python-lib/_cm.json
@@ -119,7 +119,8 @@
     },
     "torch": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torch"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torch",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torch_cuda": {
@@ -138,7 +139,8 @@
     },
     "torchvision": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchvision"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchvision",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torchvision_cuda": {
@@ -157,7 +159,8 @@
     },
     "torchaudio": {
       "env": {
-        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchaudio"
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "torchaudio",
+        "CM_GENERIC_PYTHON_PIP_EXTRA_INDEX_URL": "https://download.pytorch.org/whl/cpu"
       }
     },
     "torchaudio_cuda": {