add mock controls for experimental features
This commit is contained in:
parent
61272b9620
commit
444369e950
|
@ -47,6 +47,7 @@ from .load import (
|
|||
get_mask_filters,
|
||||
get_network_models,
|
||||
get_noise_sources,
|
||||
get_prompt_filters,
|
||||
get_source_filters,
|
||||
get_upscaling_models,
|
||||
get_wildcard_data,
|
||||
|
@ -203,10 +204,12 @@ def list_extra_strings(server: ServerContext):
|
|||
|
||||
def list_filters(server: ServerContext):
|
||||
mask_filters = list(get_mask_filters().keys())
|
||||
prompt_filters = list(get_prompt_filters().keys())
|
||||
source_filters = list(get_source_filters().keys())
|
||||
return jsonify(
|
||||
{
|
||||
"mask": mask_filters,
|
||||
"prompt": prompt_filters,
|
||||
"source": source_filters,
|
||||
}
|
||||
)
|
||||
|
|
|
@ -83,6 +83,12 @@ source_filters = {
|
|||
"segment": source_filter_segment,
|
||||
"scribble": source_filter_scribble,
|
||||
}
|
||||
prompt_filters = [
|
||||
"AUTOMATIC/promptgen-lexart",
|
||||
"AUTOMATIC/promptgen-majinai-safe",
|
||||
"AUTOMATIC/promptgen-majinai-unsafe",
|
||||
"Gustavosta/MagicPrompt-Stable-Diffusion",
|
||||
]
|
||||
|
||||
# Available ORT providers
|
||||
available_platforms: List[DeviceParams] = []
|
||||
|
@ -148,6 +154,10 @@ def get_noise_sources():
|
|||
return noise_sources
|
||||
|
||||
|
||||
def get_prompt_filters():
|
||||
return prompt_filters
|
||||
|
||||
|
||||
def get_source_filters():
|
||||
return source_filters
|
||||
|
||||
|
|
|
@ -20,6 +20,7 @@ import { UpscaleControl } from '../control/UpscaleControl.js';
|
|||
import { ImageInput } from '../input/ImageInput.js';
|
||||
import { NumericField } from '../input/NumericField.js';
|
||||
import { QueryList } from '../input/QueryList.js';
|
||||
import { ExperimentalControl } from '../control/ExperimentalControl.js';
|
||||
|
||||
export function Img2Img() {
|
||||
const { params } = mustExist(useContext(ConfigContext));
|
||||
|
@ -140,6 +141,10 @@ export function Img2Img() {
|
|||
</Stack>
|
||||
<HighresControl selectHighres={selectHighres} setHighres={setHighres} />
|
||||
<UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
|
||||
<ExperimentalControl setExperimental={(props) => {
|
||||
// eslint-disable-next-line no-console
|
||||
console.log('setting experimental props', props);
|
||||
}} />
|
||||
<Button
|
||||
disabled={doesExist(source) === false}
|
||||
variant='contained'
|
||||
|
|
|
@ -22,6 +22,7 @@ import { ImageInput } from '../input/ImageInput.js';
|
|||
import { MaskCanvas } from '../input/MaskCanvas.js';
|
||||
import { NumericField } from '../input/NumericField.js';
|
||||
import { QueryList } from '../input/QueryList.js';
|
||||
import { ExperimentalControl } from '../control/ExperimentalControl.js';
|
||||
|
||||
export function Inpaint() {
|
||||
const { params } = mustExist(useContext(ConfigContext));
|
||||
|
@ -222,6 +223,10 @@ export function Inpaint() {
|
|||
<OutpaintControl />
|
||||
<HighresControl selectHighres={selectHighres} setHighres={setHighres} />
|
||||
<UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
|
||||
<ExperimentalControl setExperimental={(props) => {
|
||||
// eslint-disable-next-line no-console
|
||||
console.log('setting experimental props', props);
|
||||
}} />
|
||||
<Button
|
||||
disabled={preventInpaint()}
|
||||
variant='contained'
|
||||
|
|
|
@ -20,6 +20,7 @@ import { ModelControl } from '../control/ModelControl.js';
|
|||
import { UpscaleControl } from '../control/UpscaleControl.js';
|
||||
import { VariableControl } from '../control/VariableControl.js';
|
||||
import { NumericField } from '../input/NumericField.js';
|
||||
import { ExperimentalControl } from '../control/ExperimentalControl.js';
|
||||
|
||||
export function SizeControl() {
|
||||
const { params } = mustExist(useContext(ConfigContext));
|
||||
|
@ -106,6 +107,10 @@ export function Txt2Img() {
|
|||
<HighresControl selectHighres={selectHighres} setHighres={setHighres} />
|
||||
<UpscaleControl selectUpscale={selectUpscale} setUpscale={setUpscale} />
|
||||
<VariableControl selectGrid={selectVariable} setGrid={setVariable} />
|
||||
<ExperimentalControl setExperimental={(props) => {
|
||||
// eslint-disable-next-line no-console
|
||||
console.log('setting experimental props', props);
|
||||
}} />
|
||||
<Button
|
||||
variant='contained'
|
||||
onClick={() => generate.mutate()}
|
||||
|
|
|
@ -44,6 +44,7 @@ export type NetworkModel = EmbeddingNetwork | LoraNetwork | ControlNetwork;
|
|||
|
||||
export interface FilterResponse {
|
||||
mask: Array<string>;
|
||||
prompt: Array<string>;
|
||||
source: Array<string>;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue