From c1900f93cd31e20b750570848bb490bf04014ca6 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Fri, 28 Jun 2024 00:39:49 +0800
Subject: [PATCH] 'update'

---
 hook_HYDiT_main.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/hook_HYDiT_main.py b/hook_HYDiT_main.py
index 8b64165..ae1eb2e 100644
--- a/hook_HYDiT_main.py
+++ b/hook_HYDiT_main.py
@@ -27,9 +27,7 @@ from transformers import BertModel, BertTokenizer, logging as tf_logging
 # from hydit.constants import VAE_EMA_PATH, TEXT_ENCODER, TOKENIZER, T5_ENCODER
 from hydit.lr_scheduler import WarmupLR
 from hydit.data_loader.arrow_load_stream import TextImageArrowStream
-from hydit.diffusion import create_diffusion
-from hydit.ds_config import deepspeed_config_from_args
-from hydit.modules.ema import EMA
+from hydit.diffusion import create_diffusion 
 from hydit.modules.fp16_layers import Float16Module
 from hydit.modules.models import HUNYUAN_DIT_MODELS
 from hydit.modules.posemb_layers import init_image_posemb
@@ -334,7 +332,7 @@ def Core(args, LOG):
     ema = None
     device = "cuda"
     if args.use_ema:
-        ema = EMA(args, model, device, logger)
+        raise ValueError("Not support EMA model.")
 
     # Setup FP16 main model:
     if args.use_fp16: