alphard/Bjornulf_custom_nodes
1
0
Fork 0
mirror of https://github.com/justUmen/Bjornulf_custom_nodes.git synced 2026-03-21 20:52:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
justumen
2025-03-25 07:53:59 +01:00
parent fbf037f66d
commit 6400396e8c
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
API_civitai.py
View File

@@ -199,7 +199,6 @@ def download_file(url, destination_path, model_name, api_token=None):
# Set up main checkpoint directory
_, civitai_base_path, parsed_models_path = get_civitai_base_paths()
bjornulf_checkpoint_path = Path(folder_paths.models_dir) / "checkpoints" / "Bjornulf_civitAI"
bjornulf_checkpoint_path.mkdir(parents=True, exist_ok=True)
# Register the main checkpoint folder
checkpoint_folders = list(folder_paths.folder_names_and_paths["checkpoints"])
@@ -781,6 +780,7 @@ class CivitAIModelSelectorSDXL:
CATEGORY = "Bjornulf"
def load_model(self, image, civitai_token):
create_bjornulf_checkpoint_folder()
if image == "none":
raise ValueError("No image selected")
@@ -883,6 +883,7 @@ class CivitAIModelSelectorFLUX_D:
CATEGORY = "Bjornulf"
def load_model(self, image, civitai_token):
create_bjornulf_checkpoint_folder()
if image == "none":
raise ValueError("No image selected")
@@ -984,6 +985,7 @@ class CivitAIModelSelectorFLUX_S:
CATEGORY = "Bjornulf"
def load_model(self, image, civitai_token):
create_bjornulf_checkpoint_folder()
if image == "none":
raise ValueError("No image selected")
@@ -1085,6 +1087,7 @@ class CivitAIModelSelectorPony:
CATEGORY = "Bjornulf"
def load_model(self, image, civitai_token):
create_bjornulf_checkpoint_folder()
if image == "none":
raise ValueError("No image selected")