Errors when updating to latest two ComfyUI?

I tried to update yesterday for Flux2, had an error when trying to use Comfy, reverted to backup all fine, tried again today for Z image, same error again.

got prompt
Using pytorch attention in VAE
Using pytorch attention in VAE
VAE load device: cuda:0, offload device: cpu, dtype: torch.bfloat16
CLIP/text encoder model load device: cuda:0, offload device: cpu, current: cpu, dtype: torch.float16
Requested to load WanTEModel
loaded completely; 22086.80 MB usable, 10835.48 MB loaded, full load: True
Requested to load WanVAE
loaded completely; 10826.52 MB usable, 242.03 MB loaded, full load: True
FETCH ComfyRegistry Data: 105/108
ComfyUI-GGUF: Allowing full torch compile
gguf qtypes: F16 (694), Q8_0 (400), F32 (1)
model weight dtype torch.float16, manual cast: None
model_type FLOW
FETCH ComfyRegistry Data [DONE]
[ComfyUI-Manager] default cache updated: https://api.comfy.org/nodes
Requested to load WAN21
FETCH DATA from: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json [DONE]
[ComfyUI-Manager] All startup tasks have been completed.
terminate called after throwing an instance of 'c10::AcceleratorError'
  what():  CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

Exception raised from c10_cuda_check_implementation at /pytorch/c10/cuda/CUDAException.cpp:42 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >) + 0x80 (0x7ff09aed9eb0 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libc10.so)
frame #1: <unknown function> + 0x111c7 (0x7ff09af6c1c7 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libc10_cuda.so)
frame #2: <unknown function> + 0x56ad1 (0x7ff09afb1ad1 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libc10_cuda.so)
frame #3: <unknown function> + 0x44166f (0x7ff08db3e66f in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libtorch_python.so)
frame #4: c10::TensorImpl::copy_tensor_metadata_except_version_counter(c10::TensorImpl const*, c10::TensorImpl*, bool) + 0x1b5 (0x7ff09aeb85e5 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libc10.so)
frame #5: c10::TensorImpl::copy_tensor_metadata(c10::TensorImpl const*, c10::TensorImpl*, c10::VariableVersion const&, bool) + 0x14 (0x7ff09aeb8614 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libc10.so)
frame #6: <unknown function> + 0x5394f88 (0x7ff07e5d9f88 in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libtorch_cpu.so)
frame #7: <unknown function> + 0x6ddf4a (0x7ff08dddaf4a in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/lib/libtorch_python.so)
<omitting python frames>
frame #14: python() [0x5581df]
frame #15: python() [0x5579ce]
frame #21: python() [0x5581df]
frame #22: python() [0x5579ce]
frame #25: python() [0x5e0c8a]
frame #27: python() [0x5e0c8a]
frame #29: python() [0x5e0c8a]
frame #31: python() [0x5e0c8a]
frame #33: python() [0x5e0c8a]
frame #34: <unknown function> + 0x79fb (0x7ff09d0319fb in /home/comfyuser/miniconda3/envs/comfyui/lib/python3.11/lib-dynload/_asyncio.cpython-311-x86_64-linux-gnu.so)
frame #36: python() [0x4cc53e]
frame #37: python() [0x51eb47]
frame #41: python() [0x5581df]
frame #42: python() [0x557a20]
frame #43: python() [0x62a8a3]
frame #44: python() [0x5fa3c4]
frame #45: <unknown function> + 0x9caa4 (0x7ff09df4aaa4 in /lib/x86_64-linux-gnu/libc.so.6)
frame #46: <unknown function> + 0x129c6c (0x7ff09dfd7c6c in /lib/x86_64-linux-gnu/libc.so.6)

Aborted (core dumped)
(comfyui) comfyuser@DA2:~/ComfyUI$

This is a WAN2.2 workflow that otherwise worked fine before the update.

Clearly some error with the GPU and CUDA or something?

I run in WSL Ubuntu and do update via manager, then the requirements.txt

Clearly something isn't right somewhere? Is the update unable to alter something in the conda environment?

It's very frustrating because when you update it just shoots through a list and doesn't really give you much opportunity to do/check anything, no warnings are shown about incompatibilities and allow you to roll back.

I click the button, and that's it, an hour to revert from backup.

It's not the end of the world but this process is so frustrating.

Setting up WSL/Ubuntu/conda and all the associated bits and pieces is a real pita (well it was 6 months ago)

Is it easier today to do a WSL deployment that doesn't take hours of painstaking CLI inputs and downloading specific files from Nvidia?

Also one last thing, it keeps complaining about Torch Audio missing. I don't have it installed. Is that likely to be causing issues?

Many thanks for any help!