nodes/Stability-AI-ComfyUI-SAI_API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix space/underscore

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-04-17 22:39:00 -07:00
parent cfa170d2b9
commit d8cfe3327f
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stability_api.py
+10 -10
View File
@@ -60,13 +60,13 @@ class StabilityBase:
"Authorization": API_KEY,
}
if kwargs.get("api_key override"):
if kwargs.get("api_key_override"):
headers = {
"Authorization": kwargs.get("api_key override"),
"Authorization": kwargs.get("api_key_override"),
}
if headers.get("Authorization") is None:
raise Exception(f"No Stability key set.\n\nUse your Stability AI API key by:\n1. Setting the SAI_API_KEY environment variable to your API key\n3. Placing inside sai_platform_key.txt\n4. Passing the API key as an argument to the function with the key 'api_key override'")
raise Exception(f"No Stability key set.\n\nUse your Stability AI API key by:\n1. Setting the SAI_API_KEY environment variable to your API key\n3. Placing inside sai_platform_key.txt\n4. Passing the API key as an argument to the function with the key 'api_key_override'")
headers["Accept"] = self.ACCEPT
@@ -108,7 +108,7 @@ class StabilityBase:
else:
error_info = response.json()
if error_info.get("name") == "unauthorized":
raise Exception(f"Stability API Error: Unauthorized.\n\nUse your Stability AI API key by:\n1. Setting the SAI_API_KEY environment variable to your API key\n3. Placing inside sai_platform_key.txt\n4. Passing the API key as an argument to the function with the key 'api_key override'")
raise Exception(f"Stability API Error: Unauthorized.\n\nUse your Stability AI API key by:\n1. Setting the SAI_API_KEY environment variable to your API key\n3. Placing inside sai_platform_key.txt\n4. Passing the API key as an argument to the function with the key 'api_key_override'")
else:
raise Exception(f"Stability API Error: {error_info}")
@@ -142,7 +142,7 @@ class StabilityCore(StabilityBase):
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"output_format": (["png", "webp", "jpeg"],),
"aspect_ratio": (["16:9", "1:1", "21:9", "2:3", "3:2", "4:5", "5:4", "9:16", "9:21"],),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
},
}
@@ -161,7 +161,7 @@ class StabilityCreativeUpscale(StabilityBase):
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"creativity": ("FLOAT", {"default": 0.3, "min": 0.01, "max": 0.35, "step": 0.01}),
"output_format": (["png", "webp", "jpeg"],),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
}
}
@@ -186,7 +186,7 @@ class StabilityInpainting(StabilityBase):
"negative_prompt": ("STRING", {"multiline": True, "default": "a post-apocalyptic world"}),
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"output_format": (["png", "webp", "jpeg"],),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
},
}
def _get_files(self, buffered, **kwargs):
@@ -215,7 +215,7 @@ class StabilitySearchAndReplace(StabilityBase):
"optional": {
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"negative_prompt": ("STRING", {"multiline": True}),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
},
}
@@ -235,7 +235,7 @@ class StabilitySD3(StabilityBase):
"image": ("IMAGE",),
"strength": ("FLOAT", {"default": 0.5, "min": 0.01, "max": 1.0, "step": 0.01}),
"aspect_ratio": (["16:9", "1:1", "21:9", "2:3", "3:2", "4:5", "5:4", "9:16", "9:21"],),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
},
}
@@ -255,6 +255,6 @@ class StabilityOutpainting(StabilityBase):
"prompt": ("STRING", {"multiline": True}),
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"output_format": (["png", "webp", "jpeg"],),
"api_key override": ("STRING", {"multiline": False}),
"api_key_override": ("STRING", {"multiline": False}),
},
}