fix the download_model is not loading the env variables

Author: ControlNet

demo_cli.py

@@ -1,10 +1,4 @@
 import asyncio
-import json
-import os
-import requests
-import time
-import tensorneko_util as N
-from pathlib import Path
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -17,12 +11,14 @@
 parser.add_argument("--base_config", type=str, required=True)
 parser.add_argument("--model_config", type=str, required=True)
 parser.add_argument("--result_folder", type=str, default="./result")
+parser.add_argument("--debug", action="store_true")
 args = parser.parse_args()
 
 from hydra_vl4ai.util.config import Config
 
 Config.base_config_path = args.base_config
 Config.model_config_path = args.model_config
+Config.debug = args.debug
 
 from hydra_vl4ai.agent.hydra import HydraNoRL
 from hydra_vl4ai.util.console import logger, console

hydra_vl4ai/_version.py

@@ -1 +1 @@
-__version__ = "0.0.5"
+__version__ = "0.0.6"

hydra_vl4ai/download_model.py

@@ -1,4 +1,7 @@
 import tensorneko_util as N
+from dotenv import load_dotenv
+
+load_dotenv()
 import argparse
 parser = argparse.ArgumentParser()
 parser.add_argument("--base_config", type=str, required=True)