Comfy-Org / Comfy-Org/ComfyUI

torch.OutOfMemoryError in KSampler Node During Sampling with WanTEModel on GTX 1050 Ti

Open
#8,540 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 10h
Merged PRs (30d)
153

Description

Encountered a torch.OutOfMemoryError while executing the KSampler node (ID: 3) in ComfyUI v0.3.38 with PyTorch 2.7.0+cu128 on a GTX 1050 Ti (4GB VRAM). The error occurs during the sampling process with the WanTEModel, specifically in the patch_embedding step. Logs indicate multiple OOM warnings, including failed VAE encoding attempts, with fallback to tiled VAE encoding. VRAM usage shows 3.48GB free out of 4GB total, suggesting potential memory management issues or model size incompatibility. The issue persists across multiple prompt executions. Please advise on optimizations or fixes for low-VRAM setups. Workflow file is too large to include but can be uploaded manually.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.