add tile size to grid jobs
This commit is contained in:
parent
1247cb7307
commit
bf830a9032
|
@ -165,7 +165,9 @@ export interface HighresParams {
|
|||
export interface Txt2ImgStage {
|
||||
name: string;
|
||||
type: 'source-txt2img';
|
||||
params: Txt2ImgParams;
|
||||
params: Txt2ImgParams & {
|
||||
tile_size: number;
|
||||
};
|
||||
}
|
||||
|
||||
export interface Img2ImgStage {
|
||||
|
@ -180,6 +182,7 @@ export interface GridStage {
|
|||
params: {
|
||||
height: number;
|
||||
width: number;
|
||||
tile_size: number;
|
||||
};
|
||||
}
|
||||
|
||||
|
@ -187,7 +190,7 @@ export interface OutputStage {
|
|||
name: string;
|
||||
type: 'persist-disk';
|
||||
params: {
|
||||
/* none */
|
||||
tile_size: number;
|
||||
};
|
||||
}
|
||||
|
||||
|
|
|
@ -30,6 +30,8 @@ export function buildPipelineForTxt2ImgGrid(grid: PipelineGrid, model: ModelPara
|
|||
...model,
|
||||
[grid.columns.parameter]: column,
|
||||
[grid.rows.parameter]: row,
|
||||
// eslint-disable-next-line camelcase
|
||||
tile_size: 8192,
|
||||
},
|
||||
});
|
||||
|
||||
|
@ -45,13 +47,18 @@ export function buildPipelineForTxt2ImgGrid(grid: PipelineGrid, model: ModelPara
|
|||
...model,
|
||||
height: grid.rows.values.length,
|
||||
width: grid.columns.values.length,
|
||||
// eslint-disable-next-line camelcase
|
||||
tile_size: 8192,
|
||||
},
|
||||
});
|
||||
|
||||
pipeline.stages.push({
|
||||
name: 'save',
|
||||
type: 'persist-disk',
|
||||
params: {},
|
||||
params: {
|
||||
// eslint-disable-next-line camelcase
|
||||
tile_size: 8192,
|
||||
},
|
||||
});
|
||||
|
||||
return pipeline;
|
||||
|
|
Loading…
Reference in New Issue