Skip to content

Commit 170ea4f

Browse files
fix(app): add CogView4ConditioningInfo to ObjectSerializerDisk's safe_globals
needed for torch w/ weights_only=True
1 parent e5b0f8b commit 170ea4f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

invokeai/app/api/dependencies.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
from invokeai.app.services.workflow_thumbnails.workflow_thumbnails_disk import WorkflowThumbnailFileStorageDisk
4040
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
4141
BasicConditioningInfo,
42+
CogView4ConditioningInfo,
4243
ConditioningFieldData,
4344
FLUXConditioningInfo,
4445
SD3ConditioningInfo,
@@ -123,6 +124,7 @@ def initialize(
123124
SDXLConditioningInfo,
124125
FLUXConditioningInfo,
125126
SD3ConditioningInfo,
127+
CogView4ConditioningInfo,
126128
],
127129
ephemeral=True,
128130
),

0 commit comments

Comments
 (0)