Skip to content

Commit 84c14dd

Browse files
Remove automatic nightly pytorch for blackwell users on desktop (#3613)
Co-authored-by: Benjamin Lu <templu1107@proton.me>
1 parent c7e6d66 commit 84c14dd

File tree

1 file changed

+0
-20
lines changed

1 file changed

+0
-20
lines changed

src/components/install/MirrorsConfiguration.vue

-20
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ import { ModelRef, computed, onMounted, ref } from 'vue'
4545
import MirrorItem from '@/components/install/mirror/MirrorItem.vue'
4646
import { PYPI_MIRROR, PYTHON_MIRROR, UVMirror } from '@/constants/uvMirrors'
4747
import { t } from '@/i18n'
48-
import { electronAPI } from '@/utils/envUtil'
4948
import { isInChina } from '@/utils/networkUtil'
5049
import { ValidationState, mergeValidationStates } from '@/utils/validationUtil'
5150
@@ -55,17 +54,6 @@ const pythonMirror = defineModel<string>('pythonMirror', { required: true })
5554
const pypiMirror = defineModel<string>('pypiMirror', { required: true })
5655
const torchMirror = defineModel<string>('torchMirror', { required: true })
5756
58-
const isBlackwellArchitecture = ref(false)
59-
60-
const requiresNightlyPytorch = async (): Promise<boolean> => {
61-
try {
62-
return await electronAPI().isBlackwell()
63-
} catch (error) {
64-
console.error('Failed to detect Blackwell architecture:', error)
65-
return false
66-
}
67-
}
68-
6957
const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
7058
const settingId = 'Comfy-Desktop.UV.TorchInstallMirror'
7159
switch (device) {
@@ -76,13 +64,6 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
7664
fallbackMirror: TorchMirrorUrl.NightlyCpu
7765
}
7866
case 'nvidia':
79-
if (isBlackwellArchitecture.value) {
80-
return {
81-
settingId,
82-
mirror: TorchMirrorUrl.NightlyCuda,
83-
fallbackMirror: TorchMirrorUrl.NightlyCuda
84-
}
85-
}
8667
return {
8768
settingId,
8869
mirror: TorchMirrorUrl.Cuda,
@@ -101,7 +82,6 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
10182
const userIsInChina = ref(false)
10283
onMounted(async () => {
10384
userIsInChina.value = await isInChina()
104-
isBlackwellArchitecture.value = await requiresNightlyPytorch()
10585
})
10686
10787
const useFallbackMirror = (mirror: UVMirror) => ({

0 commit comments

Comments
 (0)