add input check

app.py

@@ -12,7 +12,7 @@ import spaces
 import torch
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
 from PIL import Image
-from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 from prompt_check import is_unsafe_prompt
 
@@ -114,7 +114,7 @@ def load_models(model_path, enable_compile=False, attention_backend="native"):
             use_auth_token=use_auth_token,
         )
 
-        text_encoder = AutoModel.from_pretrained(
+        text_encoder = AutoModelForCausalLM.from_pretrained(
             f"{model_path}",
             subfolder="text_encoder",
             torch_dtype=torch.bfloat16,