Merge branch 'main' of github.com:tenstorrent/tt-studio into staging (#…

…149)

app/api/shared_config/model_config.py

@@ -59,6 +59,18 @@ def __post_init__(self):
         self.docker_config["environment"]["HF_HOME"] = Path(
             backend_config.model_container_cache_root
         ).joinpath("huggingface")
+
+        # Set environment variable if N150 or N300x4 is in the device configurations
+        if DeviceConfigurations.N150 in self.device_configurations or DeviceConfigurations.N300x4 in self.device_configurations:
+            self.docker_config["environment"]["WH_ARCH_YAML"] = "wormhole_b0_80_arch_eth_dispatch.yaml"
+
+        if self.env_file:
+            logger.info(f"Using env file: {self.env_file}")
+            # env file should be in persistent volume mounted
+            env_dict = load_dotenv_dict(self.env_file)
+            # env file overrides any existing docker environment variables
+            self.docker_config["environment"].update(env_dict)
+
 
         # Set environment variable if N150_WH_ARCH_YAML or N300x4_WH_ARCH_YAML is in the device configurations
         if (