nodes/MrForExample-ComfyUI-3D-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated code for diffusers fix charactergen

Browse Source
This commit is contained in:
Denis Kochetov
2025-08-07 08:49:13 +00:00
parent ca83e19986
commit 51189f74ff
4 changed files with 23 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py
+5 -4
View File
@@ -1417,7 +1417,7 @@ class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixi
return_unused_kwargs=True,
return_commit_hash=True,
force_download=force_download,
resume_download=resume_download,
# resume_download=resume_download,
proxies=proxies,
local_files_only=local_files_only,
use_auth_token=use_auth_token,
@@ -1476,7 +1476,7 @@ class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixi
weights_name=_add_variant(SAFETENSORS_WEIGHTS_NAME, variant),
cache_dir=cache_dir,
force_download=force_download,
resume_download=resume_download,
# resume_download=resume_download,
proxies=proxies,
local_files_only=local_files_only,
use_auth_token=use_auth_token,
@@ -1495,7 +1495,7 @@ class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixi
weights_name=_add_variant(WEIGHTS_NAME, variant),
cache_dir=cache_dir,
force_download=force_download,
resume_download=resume_download,
# resume_download=resume_download,
proxies=proxies,
local_files_only=local_files_only,
use_auth_token=use_auth_token,
@@ -1507,7 +1507,8 @@ class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixi
model = cls.from_config(config, **unused_kwargs)
import copy
state_dict_pretrain = load_state_dict(model_file, variant=variant)
# state_dict_pretrain = load_state_dict(model_file, variant=variant)
state_dict = load_state_dict(model_file)
state_dict = copy.deepcopy(state_dict_pretrain)
if init_mvattn_with_selfattn: