zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 10:03:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add SeedVR2 support (CORE-6) (#14110)

Browse Source
This commit is contained in:
John Pollock
2026-06-08 05:15:05 -05:00
committed by GitHub
parent 739061dd4c
commit 7863cf0e53
26 changed files with 7383 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/supported_models_base.py
View File

@@ -115,7 +115,7 @@ class BASE:
replace_prefix = {"": self.vae_key_prefix[0]}
return utils.state_dict_prefix_replace(state_dict, replace_prefix)
def set_inference_dtype(self, dtype, manual_cast_dtype):
def set_inference_dtype(self, dtype, manual_cast_dtype, device=None):
self.unet_config['dtype'] = dtype
self.manual_cast_dtype = manual_cast_dtype