1580158
/
Auto-Photoshop-StableDiffusion-Plugin
mirror of https://github.com/AbdullahAlfaraj/Auto-Photoshop-StableDiffusion-Plugin.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor controlnet mode options

pull/330/head
Abdullah Alfaraj 2023-07-12 21:41:33 +03:00
parent 33157603d1
commit 0e382c60eb
2 changed files with 39 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
typescripts/controlnet/ControlNetUnit.tsx
View File

@ -616,27 +616,42 @@ export default class ControlNetUnit extends React.Component<
</SpCheckBox>
{this.props.appState.controlnetApiVersion > 1 && (
<sp-radio-group
id={`rgControlNetMode_${this.props.index}`}
style={{ display: 'flex' }}
selected={
this.props.appState.controlNetUnitData[
this.props.index
].control_mode
}
onClick={(event: any) => {
this.props.appState.controlNetUnitData[
this.props.index
].control_mode = event.target.value
}}
>
<sp-label slot="label">
{Locale('Control Mode')}
</sp-label>
<sp-radio checked value="0">
{Locale('Balanced')}
</sp-radio>
<sp-radio
title="My prompt is more important"
value="1"
>
{Locale('My prompt is more important')}
</sp-radio>
<sp-radio
title="ControlNet is more important"
value="2"
>
{Locale('ControlNet is more important')}
</sp-radio>
{controlnetModes.map(
(mode: ControlnetMode, index: number) => {
console.log('mode:', mode, ' index:', index)
return (
<sp-radio
key={`mode-${index}`}
checked={
this.props.appState
.controlNetUnitData[
this.props.index
].control_mode === mode
? true
: void 0
}
value={`${mode}`}
>
{Locale(mode)}
</sp-radio>
)
}
)}
</sp-radio-group>
)}

12
typescripts/controlnet/store.ts
View File

@ -23,6 +23,13 @@ export const DefaultControlNetUnitData = {
auto_image: true,
}
export const controlnetModes = [
'Balanced',
'My prompt is more important',
'ControlNet is more important',
] as const
export type ControlnetMode = (typeof controlnetModes)[number]
export interface controlNetUnitData {
enabled: boolean
input_image: string
@ -44,10 +51,7 @@ export interface controlNetUnitData {
guidance_end: number
guessmode: boolean
control_mode:
| 'Balanced'
| 'My prompt is more important'
| 'ControlNet is more important'
control_mode: ControlnetMode
pixel_perfect: boolean
auto_image: boolean // sync CtrlNet image with sd input image
}