Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

add nvidia-cosmos-predict-2

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-06-26 12:21:35 -04:00
parent 61e949c1cc
commit ada38d57b4
13 changed files with 150 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+1 -1
View File
@@ -130,7 +130,7 @@ def full_vae_decode(latents, model):
# normalize latents
latents_mean = model.vae.config.get("latents_mean", None)
latents_std = model.vae.config.get("latents_std", None)
scaling_factor = model.vae.config.get("scaling_factor", None)
scaling_factor = model.vae.config.get("scaling_factor", 1.0)
shift_factor = model.vae.config.get("shift_factor", None)
if latents_mean and latents_std:
latents_mean = (torch.tensor(latents_mean).view(1, -1, 1, 1).to(latents.device, latents.dtype))