xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-23 20:04:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-09-19 01:26:43 +03:00
parent a125f61aad
commit 14515acd4a
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@ -95,10 +95,12 @@ class DownloadAndLoadCogVideoModel:
for name, param in transformer.named_parameters(): for name, param in transformer.named_parameters():
if name != "pos_embedding": if name != "pos_embedding":
param.data = param.data.to(torch.float8_e4m3fn) param.data = param.data.to(torch.float8_e4m3fn)
else: elif "I2V" in model:
for name, param in transformer.named_parameters(): for name, param in transformer.named_parameters():
if "patch_embed" not in name: if "patch_embed" not in name:
param.data = param.data.to(torch.float8_e4m3fn) param.data = param.data.to(torch.float8_e4m3fn)
else:
transformer.to(torch.float8_e4m3fn)
if fp8_transformer == "fastmode": if fp8_transformer == "fastmode":
from .fp8_optimization import convert_fp8_linear from .fp8_optimization import convert_fp8_linear
@ -557,8 +559,7 @@ class CogVideoXFunSampler:
original_height = opt_empty_latent["samples"][0].shape[-2] * 8 original_height = opt_empty_latent["samples"][0].shape[-2] * 8
closest_size, closest_ratio = get_closest_ratio(original_height, original_width, ratios=aspect_ratio_sample_size) closest_size, closest_ratio = get_closest_ratio(original_height, original_width, ratios=aspect_ratio_sample_size)
height, width = [int(x / 16) * 16 for x in closest_size] height, width = [int(x / 16) * 16 for x in closest_size]
print(f"Closest size: {width}:{height}") print(f"Closest size: {width}x{height}")
base_path = pipeline["base_path"] base_path = pipeline["base_path"]