Update llm_inference.py

llm_inference.py

@@ -22,6 +22,24 @@ class LLMInferenceNode:
             api_key=self.sambanova_api_key,
             base_url="https://api.sambanova.ai/v1",
         )
+        self.download_models()
+
+    def download_models(self)
+        from huggingface_hub import hf_hub_download
+    
+        hf_hub_download(
+            repo_id="stabilityai/stable-diffusion-3.5-large",
+            filename="mmdit.png",
+            local_dir = "./models",
+            token = self.huggingface_token
+        )
+    
+        hf_hub_download(
+            repo_id="stabilityai/stable-diffusion-3.5-large-turbo",
+            filename="LICENSE.md",
+            local_dir = "./models",
+            token = self.huggingface_token
+        )
 
     def generate_prompt(self, dynamic_seed, prompt_type, custom_input):
         """