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

...

Browse Source
This commit is contained in:
justumen
2025-03-25 07:56:29 +01:00
parent 6400396e8c
commit 46a8ddfb95
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
API_civitai.py
View File

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