-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_classes.ps1
57 lines (46 loc) · 1.69 KB
/
generate_classes.ps1
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#!/usr/bin/env pwsh
#
# This powershell script is NOT for any release, and has no end-user value.
#
Set-StrictMode -Version Latest
$WORKFLOWS_ALL = @(
"comfyui_default",
"img2img_sdxl_0.3",
"inpainting_sdxl_0.4",
"sytan_sdxl_1.0"
)
$GENERATORS_ALL = @(
"generate_node_accessor.py"
"generate_inputs_dialog.py"
)
function Get-APIJsons([string[]]$SubjectWorkflows) {
$api_jsons = @()
foreach ($workflow in $SubjectWorkflows) {
$api_jsons += "{0}_workflow_api.json" -f $workflow
}
return $api_jsons
}
function Get-APIJsonsMap([string[]]$SubjectWorkflows) {
$api_jsons_map = @{}
foreach ($workflow in $SubjectWorkflows) {
$api_jsons_map.add($workflow,"{0}_workflow_api.json" -f $workflow)
}
return $api_jsons_map
}
function Generate-Classes([String] $generator, [string[]]$SubjectWorkflows) {
$wfmap = Get-APIJsonsMap($SubjectWorkflows)
foreach ($workflow in $SubjectWorkflows) {
python $generator $wfmap[$workflow]
}
}
Set-Python 11
$ENV:PYTHONPATH=$PSScriptRoot
foreach($generator_name in @("generate_inputs_dialog.py")){
$generator = Join-Path -Path $PSScriptRoot -ChildPath "workflow" -AdditionalChildPath @($generator_name)
Generate-Classes $generator $WORKFLOWS_ALL
# Generate-Classes $generator @("sytan_sdxl_1.0")
}
# Test with
# Remove-Item $ENV:TEMP\GimpComfyUI_logfile.txt; gimp3 "M:\stills\A_I\misc\example.png" "M:\stills\A_I\misc\1962_TR3B_scaled.png" "M:\stills\A_I\misc\1962_TR3B_mask_scaled.png" "M:\stills\A_I\misc\Kuo-toa_05.png"
# or
# rm -f $TMPDIR/GimpComfyUI_logfile.txt; gimp3 ~/Pictures/example.png ~/Pictures/1962_TR3B_scaled.png ~/Pictures/1962_TR3B_mask_scaled.png ~/Pictures/Kuo-toa_05.png