FilesExpand file tree

master

/

choose_upscale_model_node.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

30 lines (25 loc) · 822 Bytes

master

/

choose_upscale_model_node.py

Copy path

Top

File metadata and controls

• Code
• Blame

30 lines (25 loc) · 822 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

class ChooseUpscaleModel:

@classmethod

def INPUT_TYPES(cls):

return {

"required": {

"upscale_model_1": ("UPSCALE_MODEL",),

"upscale_model_2": ("UPSCALE_MODEL",),

"use_model_1": ("BOOLEAN", {"default": True}),

}

}

RETURN_TYPES = ("UPSCALE_MODEL",)

RETURN_NAMES = ("upscale_model",)

FUNCTION = "choose_upscale_model"

CATEGORY = "ControlAltAI Nodes/Logic"

def choose_upscale_model(self, upscale_model_1, upscale_model_2, use_model_1):

if use_model_1:

return (upscale_model_1,)

else:

return (upscale_model_2,)

NODE_CLASS_MAPPINGS = {

"ChooseUpscaleModel": ChooseUpscaleModel,

}

NODE_DISPLAY_NAME_MAPPINGS = {

"ChooseUpscaleModel": "Choose Upscale Model",

}