Skip to content

GitLab

Unverified Commit 1130801c authored by Muyang Li's avatar Muyang Li Committed by GitHub
Browse files

fix: fix the controlnet argument errors (#376)

parent 140b21e5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nunchaku/__version__.py
View file @ 1130801c
__version__ = "0.3.0dev1" __version__ = "0.3.0dev2"
nunchaku/caching/utils.py
View file @ 1130801c
...@@ -399,7 +399,7 @@ class FluxCachedTransformerBlocks(nn.Module): ...@@ -399,7 +399,7 @@ class FluxCachedTransformerBlocks(nn.Module):
controlnet_block_samples = ( controlnet_block_samples = (
torch.stack(controlnet_block_samples).to(original_device) if len(controlnet_block_samples) > 0 else None torch.stack(controlnet_block_samples).to(original_device) if len(controlnet_block_samples) > 0 else None
) )
if controlnet_single_block_samples is not None and len(controlnet_single_block_samples) > 0: if controlnet_single_block_samples is not None:
controlnet_single_block_samples = ( controlnet_single_block_samples = (
torch.stack(controlnet_single_block_samples).to(original_device) torch.stack(controlnet_single_block_samples).to(original_device)
if len(controlnet_single_block_samples) > 0 if len(controlnet_single_block_samples) > 0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment