yhayano-ponotech
0c581b628a
- Add Flux1.1[pro] node - Change API end point for Flux1.0[pro] to "fal-ai/flux-pro/new"
28 lines
808 B
Python
28 lines
808 B
Python
from .base_fal_api_flux_node import BaseFalAPIFluxNode
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class FalAPIFluxProNode(BaseFalAPIFluxNode):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.set_api_endpoint("fal-ai/flux-pro/new")
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
input_types = super().INPUT_TYPES()
|
|
input_types["required"]["safety_tolerance"] = (["1", "2", "3", "4", "5", "6"],)
|
|
return input_types
|
|
|
|
def prepare_arguments(self, safety_tolerance, **kwargs):
|
|
arguments = super().prepare_arguments(**kwargs)
|
|
arguments["safety_tolerance"] = safety_tolerance
|
|
return arguments
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FalAPIFluxProNode": FalAPIFluxProNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FalAPIFluxProNode": "Fal API Flux Pro"
|
|
} |