tidy(ui): rename generation tab graph builders

This commit is contained in:
psychedelicious 2024-05-03 10:05:43 +10:00 committed by Kent Keirsey
parent e354fee4f4
commit 4c7be03702
3 changed files with 7 additions and 7 deletions

View File

@ -1,8 +1,8 @@
import { enqueueRequested } from 'app/store/actions'; import { enqueueRequested } from 'app/store/actions';
import type { AppStartListening } from 'app/store/middleware/listenerMiddleware'; import type { AppStartListening } from 'app/store/middleware/listenerMiddleware';
import { buildGenerationTabGraph } from 'features/nodes/util/graph/buildGenerationTabGraph';
import { buildGenerationTabSDXLGraph } from 'features/nodes/util/graph/buildGenerationTabSDXLGraph';
import { prepareLinearUIBatch } from 'features/nodes/util/graph/buildLinearBatchConfig'; import { prepareLinearUIBatch } from 'features/nodes/util/graph/buildLinearBatchConfig';
import { buildLinearSDXLTextToImageGraph } from 'features/nodes/util/graph/buildLinearSDXLTextToImageGraph';
import { buildLinearTextToImageGraph } from 'features/nodes/util/graph/buildLinearTextToImageGraph';
import { queueApi } from 'services/api/endpoints/queue'; import { queueApi } from 'services/api/endpoints/queue';
export const addEnqueueRequestedLinear = (startAppListening: AppStartListening) => { export const addEnqueueRequestedLinear = (startAppListening: AppStartListening) => {
@ -17,9 +17,9 @@ export const addEnqueueRequestedLinear = (startAppListening: AppStartListening)
let graph; let graph;
if (model && model.base === 'sdxl') { if (model && model.base === 'sdxl') {
graph = await buildLinearSDXLTextToImageGraph(state); graph = await buildGenerationTabSDXLGraph(state);
} else { } else {
graph = await buildLinearTextToImageGraph(state); graph = await buildGenerationTabGraph(state);
} }
const batchConfig = prepareLinearUIBatch(state, graph, prepend); const batchConfig = prepareLinearUIBatch(state, graph, prepend);

View File

@ -25,7 +25,7 @@ import {
} from './constants'; } from './constants';
import { addCoreMetadataNode, getModelMetadataField } from './metadata'; import { addCoreMetadataNode, getModelMetadataField } from './metadata';
export const buildLinearTextToImageGraph = async (state: RootState): Promise<NonNullableGraph> => { export const buildGenerationTabGraph = async (state: RootState): Promise<NonNullableGraph> => {
const log = logger('nodes'); const log = logger('nodes');
const { const {
model, model,

View File

@ -25,7 +25,7 @@ import {
import { getBoardField, getIsIntermediate, getSDXLStylePrompts } from './graphBuilderUtils'; import { getBoardField, getIsIntermediate, getSDXLStylePrompts } from './graphBuilderUtils';
import { addCoreMetadataNode, getModelMetadataField } from './metadata'; import { addCoreMetadataNode, getModelMetadataField } from './metadata';
export const buildLinearSDXLTextToImageGraph = async (state: RootState): Promise<NonNullableGraph> => { export const buildGenerationTabSDXLGraph = async (state: RootState): Promise<NonNullableGraph> => {
const log = logger('nodes'); const log = logger('nodes');
const { const {
model, model,
@ -224,7 +224,7 @@ export const buildLinearSDXLTextToImageGraph = async (state: RootState): Promise
addCoreMetadataNode( addCoreMetadataNode(
graph, graph,
{ {
generation_mode: 'sdxl_txt2img', generation_mode: 'txt2img',
cfg_scale, cfg_scale,
cfg_rescale_multiplier, cfg_rescale_multiplier,
height, height,