enable hires fix in all comfyui modes

pull/369/head
Abdullah Alfaraj 2023-11-19 03:03:43 +03:00
parent 3fd3a4fc9d
commit 8d73846cbf
2 changed files with 18 additions and 8 deletions

View File

@ -38,6 +38,8 @@ import {
onHeightSliderInput, onHeightSliderInput,
heightSliderOnChangeEventHandler, heightSliderOnChangeEventHandler,
loadPresetSettings, loadPresetSettings,
isHiResMode,
} from './util' } from './util'
import { general } from '../util/oldSystem' import { general } from '../util/oldSystem'
import { requestSwapModel, setInpaintMaskWeight } from '../util/ts/sdapi' import { requestSwapModel, setInpaintMaskWeight } from '../util/ts/sdapi'
@ -1103,11 +1105,7 @@ class SDTab extends React.Component<{}> {
class="checkbox" class="checkbox"
id="chHiResFixs" id="chHiResFixs"
style={{ style={{
display: [ScriptMode.Txt2Img].includes( display: isHiResMode() ? 'flex' : 'none',
store.data.rb_mode
)
? 'flex'
: 'none',
}} }}
checked={store.data.enable_hr} checked={store.data.enable_hr}
onClick={(evt: any) => { onClick={(evt: any) => {
@ -1131,9 +1129,7 @@ class SDTab extends React.Component<{}> {
id="HiResDiv" id="HiResDiv"
style={{ style={{
display: display:
[ScriptMode.Txt2Img].includes( isHiResMode() && store.data.enable_hr
store.data.rb_mode
) && store.data.enable_hr
? void 0 ? void 0
: 'none', : 'none',
}} }}

View File

@ -714,6 +714,20 @@ export function loadPresetSettings(preset: any) {
// io_ts.presetToStore(preset?.controlnet_tab_preset, store) // io_ts.presetToStore(preset?.controlnet_tab_preset, store)
} }
} }
export function isHiResMode() {
let is_hi_res_mode = false
if (settings_tab_ts.store.data.selected_backend === 'Automatic1111') {
is_hi_res_mode = [ScriptMode.Txt2Img].includes(store.data.rb_mode)
} else if (settings_tab_ts.store.data.selected_backend === 'ComfyUI') {
is_hi_res_mode = [
ScriptMode.Txt2Img,
ScriptMode.Img2Img,
ScriptMode.Inpaint,
ScriptMode.Outpaint,
].includes(store.data.rb_mode)
}
return is_hi_res_mode
}
export default { export default {
store: store, store: store,