[nit] Use TorchMirrorUrl (#3617)

Co-authored-by: Benjamin Lu <templu1107@proton.me>
This commit is contained in:
Benjamin Lu
2025-04-25 10:52:52 -04:00
committed by GitHub
parent b79cbf69af
commit 1c59e3b51b

View File

@@ -35,8 +35,6 @@
<script setup lang="ts">
import {
CUDA_TORCH_URL,
NIGHTLY_CPU_TORCH_URL,
TorchDeviceType,
TorchMirrorUrl
} from '@comfyorg/comfyui-electron-types'
@@ -74,8 +72,8 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
case 'mps':
return {
settingId,
mirror: NIGHTLY_CPU_TORCH_URL,
fallbackMirror: NIGHTLY_CPU_TORCH_URL
mirror: TorchMirrorUrl.NightlyCpu,
fallbackMirror: TorchMirrorUrl.NightlyCpu
}
case 'nvidia':
if (isBlackwellArchitecture.value) {
@@ -87,8 +85,8 @@ const getTorchMirrorItem = (device: TorchDeviceType): UVMirror => {
}
return {
settingId,
mirror: CUDA_TORCH_URL,
fallbackMirror: CUDA_TORCH_URL
mirror: TorchMirrorUrl.Cuda,
fallbackMirror: TorchMirrorUrl.Cuda
}
case 'cpu':
default: