xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-08 20:34:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

API key from file

Browse Source
This commit is contained in:
kijai 2024-04-18 09:00:49 +03:00
parent 671af53b34
commit f7cb1a19dc
2 changed files with 36 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
config.json Normal file
View File

@ -0,0 +1,3 @@
{
"sai_api_key": "your_api_key_here"
}

40
nodes.py
View File

@ -4619,7 +4619,6 @@ class StabilityAPI_SD3:
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return { return {
"required": { "required": {
"api_key": ("STRING", {"multiline": True}),
"prompt": ("STRING", {"multiline": True}), "prompt": ("STRING", {"multiline": True}),
"n_prompt": ("STRING", {"multiline": True}), "n_prompt": ("STRING", {"multiline": True}),
"seed": ("INT", {"default": 123,"min": 0, "max": 4294967294, "step": 1}), "seed": ("INT", {"default": 123,"min": 0, "max": 4294967294, "step": 1}),
@ -4656,9 +4655,11 @@ class StabilityAPI_SD3:
}), }),
}, },
"optional": { "optional": {
"api_key": ("STRING", {"multiline": True}),
"image": ("IMAGE",), "image": ("IMAGE",),
"img2img_strength": ("FLOAT", {"default": 0.5, "min": 0.0, "max": 1.0, "step": 0.01}), "img2img_strength": ("FLOAT", {"default": 0.5, "min": 0.0, "max": 1.0, "step": 0.01}),
} "disable_metadata": ("BOOLEAN", {"default": True}),
},
} }
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
@ -4672,6 +4673,13 @@ Although you may have multiple keys in your account,
you should use the same key for all requests to this API. you should use the same key for all requests to this API.
Get your API key here: https://platform.stability.ai/account/keys Get your API key here: https://platform.stability.ai/account/keys
Recommended to set the key in the config.json -file under this
node packs folder.
# WARNING:
Otherwise the API key may get saved in the image metadata even
with "disable_metadata" on if the workflow includes save nodes
separate from this node.
sd3 requires 6.5 credits per generation sd3 requires 6.5 credits per generation
sd3-turbo requires 4 credits per generation sd3-turbo requires 4 credits per generation
@ -4679,8 +4687,13 @@ If no image is provided, mode is set to text-to-image
""" """
def apicall(self, api_key, prompt, n_prompt, model, seed, aspect_ratio, output_format, def apicall(self, prompt, n_prompt, model, seed, aspect_ratio, output_format,
img2img_strength=0.5, image=None): img2img_strength=0.5, image=None, disable_metadata=True, api_key=""):
from comfy.cli_args import args
if disable_metadata:
args.disable_metadata = True
else:
args.disable_metadata = False
import requests import requests
from io import BytesIO from io import BytesIO
@ -4714,12 +4727,23 @@ If no image is provided, mode is set to text-to-image
if model != "sd3-turbo": if model != "sd3-turbo":
data["negative_prompt"] = n_prompt data["negative_prompt"] = n_prompt
headers={
"accept": "image/*"
}
if api_key != "":
headers["authorization"] = api_key
else:
config_file_path = os.path.join(script_directory,"config.json")
with open(config_file_path, 'r') as file:
config = json.load(file)
api_key_from_config = config.get("sai_api_key")
headers["authorization"] = api_key_from_config
response = requests.post( response = requests.post(
f"https://api.stability.ai/v2beta/stable-image/generate/sd3", f"https://api.stability.ai/v2beta/stable-image/generate/sd3",
headers={ headers=headers,
"authorization": api_key,
"accept": "image/*"
},
files = files, files = files,
data = data, data = data,
) )