init

app.py

@@ -17,8 +17,8 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 # --- MODIFICATION START ---
 # Import necessary model classes from the local directory
 # This assumes you have your model files in a 'llada_code' subfolder
-from llada_code.modeling_llada import LLaDAModelLM
-from llada_code.configuration_llada import LLaDAConfig
+from model_cache.llada.modeling_llada import LLaDAModelLM
+from model_cache.llada.configuration_llada import LLaDAConfig
 # --- MODIFICATION END ---