Merge branch 'main' into vap
This commit is contained in:
+1
-1
@@ -195,7 +195,7 @@ class WanVideoSampler:
|
|||||||
block.audio_block = None
|
block.audio_block = None
|
||||||
|
|
||||||
if not transformer.patched_linear and patcher.model["sd"] is not None and len(patcher.patches) != 0 and gguf_reader is None:
|
if not transformer.patched_linear and patcher.model["sd"] is not None and len(patcher.patches) != 0 and gguf_reader is None:
|
||||||
transformer = _replace_linear(transformer, dtype, patcher.model["sd"], model["compile_args"])
|
transformer = _replace_linear(transformer, dtype, patcher.model["sd"], compile_args=model["compile_args"])
|
||||||
transformer.patched_linear = True
|
transformer.patched_linear = True
|
||||||
if patcher.model["sd"] is not None and gguf_reader is None:
|
if patcher.model["sd"] is not None and gguf_reader is None:
|
||||||
load_weights(patcher.model.diffusion_model, patcher.model["sd"], weight_dtype, base_dtype=dtype, transformer_load_device=device,
|
load_weights(patcher.model.diffusion_model, patcher.model["sd"], weight_dtype, base_dtype=dtype, transformer_load_device=device,
|
||||||
|
|||||||
+1
-1
@@ -172,7 +172,7 @@ class WanVideoDiffusionForcingSampler:
|
|||||||
|
|
||||||
# Load weights
|
# Load weights
|
||||||
if not transformer.patched_linear and patcher.model["sd"] is not None and len(patcher.patches) != 0:
|
if not transformer.patched_linear and patcher.model["sd"] is not None and len(patcher.patches) != 0:
|
||||||
transformer = _replace_linear(transformer, dtype, patcher.model["sd"], model["compile_args"])
|
transformer = _replace_linear(transformer, dtype, patcher.model["sd"], compile_args=model["compile_args"])
|
||||||
transformer.patched_linear = True
|
transformer.patched_linear = True
|
||||||
if patcher.model["sd"] is not None and gguf_reader is None:
|
if patcher.model["sd"] is not None and gguf_reader is None:
|
||||||
load_weights(patcher.model.diffusion_model, patcher.model["sd"], weight_dtype, base_dtype=dtype, transformer_load_device=device, block_swap_args=block_swap_args)
|
load_weights(patcher.model.diffusion_model, patcher.model["sd"], weight_dtype, base_dtype=dtype, transformer_load_device=device, block_swap_args=block_swap_args)
|
||||||
|
|||||||
Reference in New Issue
Block a user