Remove automatic nightly pytorch for blackwell users on desktop (#3613)

Co-authored-by: Benjamin Lu <templu1107@proton.me>
This commit is contained in:
Benjamin Lu
2025-04-25 18:08:37 -04:00
committed by GitHub
parent c7e6d66d47
commit 84c14ddd92

View File

@@ -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) => ({