Commit
•
476c6c0
1
Parent(s):
e4a3887
Update llm_inference.py
Browse files- llm_inference.py +1 -1
llm_inference.py
CHANGED
|
@@ -22,7 +22,7 @@ class LLMInferenceNode:
|
|
| 22 |
api_key=self.sambanova_api_key,
|
| 23 |
base_url="https://api.sambanova.ai/v1",
|
| 24 |
)
|
| 25 |
-
self.flux_client = Client("KingNish/Realtime-FLUX"
|
| 26 |
|
| 27 |
def generate_prompt(self, dynamic_seed, prompt_type, custom_input):
|
| 28 |
"""
|
|
|
|
| 22 |
api_key=self.sambanova_api_key,
|
| 23 |
base_url="https://api.sambanova.ai/v1",
|
| 24 |
)
|
| 25 |
+
self.flux_client = Client("KingNish/Realtime-FLUX", hf_token=self.huggingface_token)
|
| 26 |
|
| 27 |
def generate_prompt(self, dynamic_seed, prompt_type, custom_input):
|
| 28 |
"""
|