[Desktop] Apply default pypi mirror when installing CPU torch (#2367)

src/components/install/MirrorsConfiguration.vue

@@ -5,10 +5,7 @@
     :collapsed="!showMirrorInputs"
     pt:root="bg-neutral-800 border-none w-[600px]"
   >
-    <template
-      v-for="([item, modelValue], index) in mirrors"
-      :key="item.settingId"
-    >
+    <template v-for="([item, modelValue], index) in mirrors" :key="item.mirror">
       <Divider v-if="index > 0" />
 
       <MirrorItem
@@ -34,25 +31,59 @@
 </template>
 
 <script setup lang="ts">
-import _ from 'lodash'
+import {
+  CUDA_TORCH_URL,
+  NIGHTLY_CPU_TORCH_URL,
+  TorchDeviceType
+} from '@comfyorg/comfyui-electron-types'
 import Divider from 'primevue/divider'
 import Panel from 'primevue/panel'
-import { computed, ref } from 'vue'
+import { ModelRef, computed, ref } from 'vue'
 
 import MirrorItem from '@/components/install/mirror/MirrorItem.vue'
-import { UV_MIRRORS } from '@/constants/uvMirrors'
+import { PYPI_MIRROR, PYTHON_MIRROR, UVMirror } from '@/constants/uvMirrors'
 import { t } from '@/i18n'
 import { ValidationState, mergeValidationStates } from '@/utils/validationUtil'
 
 const showMirrorInputs = ref(false)
+const { device } = defineProps<{ device: TorchDeviceType }>()
 const pythonMirror = defineModel<string>('pythonMirror', { required: true })
 const pypiMirror = defineModel<string>('pypiMirror', { required: true })
 const torchMirror = defineModel<string>('torchMirror', { required: true })
 
-const mirrors = _.zip(UV_MIRRORS, [pythonMirror, pypiMirror, torchMirror])
+const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
+  const settingId = 'Comfy-Desktop.UV.TorchInstallMirror'
+  switch (device) {
+    case 'mps':
+      return {
+        settingId,
+        mirror: NIGHTLY_CPU_TORCH_URL,
+        fallbackMirror: NIGHTLY_CPU_TORCH_URL
+      }
+    case 'nvidia':
+      return {
+        settingId,
+        mirror: CUDA_TORCH_URL,
+        fallbackMirror: CUDA_TORCH_URL
+      }
+    case 'cpu':
+    default:
+      return {
+        settingId,
+        mirror: PYPI_MIRROR.mirror,
+        fallbackMirror: PYPI_MIRROR.fallbackMirror
+      }
+  }
+}
+
+const mirrors = computed<[UVMirror, ModelRef<string>][]>(() => [
+  [PYTHON_MIRROR, pythonMirror],
+  [PYPI_MIRROR, pypiMirror],
+  [getTorchMirrorItem(device), torchMirror]
+])
 
 const validationStates = ref<ValidationState[]>(
-  mirrors.map(() => ValidationState.IDLE)
+  mirrors.value.map(() => ValidationState.IDLE)
 )
 const validationState = computed(() => {
   return mergeValidationStates(validationStates.value)