diff --git a/huggingface_guess/model_list.py b/huggingface_guess/model_list.py
index e1e5882..fc6fcf9 100644
--- a/huggingface_guess/model_list.py
+++ b/huggingface_guess/model_list.py
@@ -632,7 +632,6 @@ class AuraFlow(BASE):
 
 class HunyuanDiT(BASE):
     unet_config = {
-        "image_model": "hydit",
     }
 
     unet_extra_config = {
@@ -652,7 +651,6 @@ class HunyuanDiT(BASE):
 
 class HunyuanDiT1(HunyuanDiT):
     unet_config = {
-        "image_model": "hydit1",
     }
 
     unet_extra_config = {}
@@ -667,7 +665,6 @@ class Flux(BASE):
     huggingface_repo = 'black-forest-labs/FLUX.1-dev'
 
     unet_config = {
-        "image_model": "flux",
         "guidance_embed": True,
     }
 
@@ -703,7 +700,6 @@ class FluxSchnell(Flux):
     huggingface_repo = 'black-forest-labs/FLUX.1-schnell'
 
     unet_config = {
-        "image_model": "flux",
         "guidance_embed": False,
     }