Remove BidirectionalLlamaAttentionLayer input

app.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer
 import os
 import importlib
 from huggingface_hub import hf_hub_download
-from llama_diffusion_model import CustomTransformerModel, CustomTransformerConfig, BidirectionalLlamaAttention, disable_dropout
+from llama_diffusion_model import CustomTransformerModel, CustomTransformerConfig, disable_dropout
 import spaces
 
 hf_token = os.getenv("HF_TOKEN")