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

0.36

Browse Source
This commit is contained in:
justumen
2024-09-17 10:15:55 +02:00
parent 31baddd065
commit 992a2e762d
4 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
random_model_clip_vae.py
View File

@@ -1,4 +1,5 @@
import random
import json
class RandomModelClipVae:
@classmethod
@@ -12,11 +13,12 @@ class RandomModelClipVae:
"model_2": ("MODEL", {"forceInput": True}),
"clip_2": ("CLIP", {"forceInput": True}),
"vae_2": ("VAE", {"forceInput": True}),
"seed": ("INT", {"default": 0}),
},
"hidden": {
**{f"model_{i}": ("MODEL", {"forceInput": True}) for i in range(3, 11)},
**{f"clip_{i}": ("CLIP", {"forceInput": True}) for i in range(3, 11)},
**{f"vae_{i}": ("VAE", {"forceInput": True}) for i in range(3, 11)}
**{f"vae_{i}": ("VAE", {"forceInput": True}) for i in range(3, 11)},
}
}
@@ -24,10 +26,11 @@ class RandomModelClipVae:
FUNCTION = "random_select"
def random_select(self, number_of_inputs, **kwargs):
random.seed(kwargs.get('seed', 0))
selected_index = random.randint(1, number_of_inputs)
selected_model = kwargs[f"model_{selected_index}"]
selected_clip = kwargs[f"clip_{selected_index}"]
selected_vae = kwargs[f"vae_{selected_index}"]
return (selected_model, selected_clip, selected_vae)
return (selected_model, selected_clip, selected_vae)