Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow passing of civit api key via config

Browse Source
This commit is contained in:
Brandon Rising
2024-01-30 20:46:42 -05:00
committed by Kent Keirsey
parent 14efc95707
commit 088e3420e6
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/install/model_install_backend.py
View File

@ -111,7 +111,7 @@ class ModelInstall(object):
self.datasets = OmegaConf.load(Dataset_path)
self.prediction_helper = prediction_type_helper
self.access_token = access_token or HfFolder.get_token()
self.civit_api_key = civit_api_key or os.environ.get("CIVIT_API_KEY")
self.civit_api_key = civit_api_key or config.civit_api_key
self.reverse_paths = self._reverse_paths(self.datasets)
def all_models(self) -> Dict[str, ModelLoadInfo]: