From 84c14ddd92e5406b8a781bf98214a02e5d5d118c Mon Sep 17 00:00:00 2001
From: Benjamin Lu <benceruleanlu@proton.me>
Date: Fri, 25 Apr 2025 18:08:37 -0400
Subject: [PATCH] Remove automatic nightly pytorch for blackwell users on
 desktop (#3613)

Co-authored-by: Benjamin Lu <templu1107@proton.me>
---
 .../install/MirrorsConfiguration.vue          | 20 -------------------
 1 file changed, 20 deletions(-)

diff --git a/src/components/install/MirrorsConfiguration.vue b/src/components/install/MirrorsConfiguration.vue
index c7ce7e5f16..0053f973a4 100644
--- a/src/components/install/MirrorsConfiguration.vue
+++ b/src/components/install/MirrorsConfiguration.vue
@@ -45,7 +45,6 @@ import { ModelRef, computed, onMounted, ref } from 'vue'
 import MirrorItem from '@/components/install/mirror/MirrorItem.vue'
 import { PYPI_MIRROR, PYTHON_MIRROR, UVMirror } from '@/constants/uvMirrors'
 import { t } from '@/i18n'
-import { electronAPI } from '@/utils/envUtil'
 import { isInChina } from '@/utils/networkUtil'
 import { ValidationState, mergeValidationStates } from '@/utils/validationUtil'
 
@@ -55,17 +54,6 @@ const pythonMirror = defineModel<string>('pythonMirror', { required: true })
 const pypiMirror = defineModel<string>('pypiMirror', { required: true })
 const torchMirror = defineModel<string>('torchMirror', { required: true })
 
-const isBlackwellArchitecture = ref(false)
-
-const requiresNightlyPytorch = async (): Promise<boolean> => {
-  try {
-    return await electronAPI().isBlackwell()
-  } catch (error) {
-    console.error('Failed to detect Blackwell architecture:', error)
-    return false
-  }
-}
-
 const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
   const settingId = 'Comfy-Desktop.UV.TorchInstallMirror'
   switch (device) {
@@ -76,13 +64,6 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
         fallbackMirror: TorchMirrorUrl.NightlyCpu
       }
     case 'nvidia':
-      if (isBlackwellArchitecture.value) {
-        return {
-          settingId,
-          mirror: TorchMirrorUrl.NightlyCuda,
-          fallbackMirror: TorchMirrorUrl.NightlyCuda
-        }
-      }
       return {
         settingId,
         mirror: TorchMirrorUrl.Cuda,
@@ -101,7 +82,6 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
 const userIsInChina = ref(false)
 onMounted(async () => {
   userIsInChina.value = await isInChina()
-  isBlackwellArchitecture.value = await requiresNightlyPytorch()
 })
 
 const useFallbackMirror = (mirror: UVMirror) => ({