HunYuanDiT missing import

2024-05-21 13:59:42 +02:00
parent ea705d06e8
commit 13dc46fffc
1 changed files with 1 additions and 0 deletions
@@ -4,6 +4,7 @@ import torch
 from transformers import AutoTokenizer, modeling_utils
 from transformers import T5Config, T5EncoderModel, BertConfig, BertModel

+from comfy import model_management
 import comfy.model_patcher
 import comfy.utils