Disable forced NVIDIA torch upgrades (#1543)

Stop forcing NVIDIA PyTorch upgrades.

Users should control large PyTorch downloads and avoid surprise
upgrades. This change removes the legacy torch mirror migration and
stops treating NVIDIA cu130 minimum versions as a required package
upgrade, so validation no longer triggers an automatic update. The
package update flow also skips the recommended NVIDIA torch install
step, keeping upgrades user-driven. Tradeoff: users can remain on older
CUDA/torch builds and may hit compatibility issues; the NVIDIA driver
warning still appears.

Author: benceruleanlu

src/install/installationManager.ts

@@ -382,7 +382,8 @@ export class InstallationManager implements HasTelemetry {
       await installation.virtualEnvironment.installComfyUIRequirements(callbacks);
       await installation.virtualEnvironment.installComfyUIManagerRequirements(callbacks);
       await this.warnIfNvidiaDriverTooOld(installation);
-      await installation.virtualEnvironment.ensureRecommendedNvidiaTorch(callbacks);
+      // Disable automatic NVIDIA torch upgrades so users control large downloads.
+      // await installation.virtualEnvironment.ensureRecommendedNvidiaTorch(callbacks);
       await installation.validate();
     } catch (error) {
       log.error('Error auto-updating packages:', error);

src/main-process/comfyInstallation.ts

@@ -3,7 +3,6 @@ import { rm } from 'node:fs/promises';
 
 import { ComfyServerConfig } from '../config/comfyServerConfig';
 import { ComfySettings, useComfySettings } from '../config/comfySettings';
-import { LEGACY_NVIDIA_TORCH_MIRROR, TorchMirrorUrl } from '../constants';
 import { evaluatePathRestrictions } from '../handlers/pathHandlers';
 import type { DesktopInstallState } from '../main_types';
 import type { InstallValidation } from '../preload';
@@ -86,28 +85,10 @@ export class ComfyInstallation {
     const basePath = config.get('basePath');
     if (state && basePath) {
       await ComfySettings.load(basePath);
-      await ComfyInstallation.migrateLegacyTorchMirrorIfNeeded();
       return new ComfyInstallation(state, basePath, getTelemetry());
     }
   }
 
-  private static async migrateLegacyTorchMirrorIfNeeded(): Promise<void> {
-    const selectedDevice = useDesktopConfig().get('selectedDevice');
-    if (selectedDevice !== 'nvidia') return;
-
-    const settings = useComfySettings();
-    const torchMirror = settings.get('Comfy-Desktop.UV.TorchInstallMirror');
-    if (!torchMirror || torchMirror !== LEGACY_NVIDIA_TORCH_MIRROR) return;
-
-    try {
-      settings.set('Comfy-Desktop.UV.TorchInstallMirror', TorchMirrorUrl.Cuda);
-      await settings.saveSettings();
-      log.info('Migrated legacy NVIDIA torch mirror from cu129 to cu130.');
-    } catch (error) {
-      log.warn('Failed to migrate legacy NVIDIA torch mirror to cu130.', error);
-    }
-  }
-
   /**
    * Validate the installation and add any results to {@link issues}.
    * @returns The validated installation state, along with a list of any issues detected.

src/virtualEnvironment.ts

@@ -11,7 +11,6 @@ import {
   AMD_ROCM_SDK_PACKAGES,
   AMD_TORCH_PACKAGES,
   InstallStage,
-  LEGACY_NVIDIA_TORCH_MIRROR,
   NVIDIA_TORCHVISION_VERSION,
   NVIDIA_TORCH_PACKAGES,
   NVIDIA_TORCH_VERSION,
@@ -115,9 +114,6 @@ function fixDeviceMirrorMismatch(device: TorchDeviceType, mirror: string | undef
     if (device === 'nvidia') return TorchMirrorUrl.Cuda;
     else if (device === 'mps') return TorchMirrorUrl.NightlyCpu;
   }
-  if (device === 'nvidia' && mirror === LEGACY_NVIDIA_TORCH_MIRROR) {
-    return TorchMirrorUrl.Cuda;
-  }
   return mirror;
 }
 
@@ -982,11 +978,6 @@ export class VirtualEnvironment implements HasTelemetry, PythonExecutor {
       return 'package-upgrade';
     }
 
-    if (await this.needsNvidiaTorchUpgrade()) {
-      log.info('NVIDIA PyTorch version out of date. Treating as package upgrade.');
-      return 'package-upgrade';
-    }
-
     log.debug('hasRequirements result:', 'OK');
     return 'OK';
   }