create batch size latents
This commit is contained in:
parent
89dbf81335
commit
10050263bf
|
@ -43,7 +43,7 @@ class SourceTxt2ImgStage(BaseStage):
|
||||||
|
|
||||||
prompt_pairs, loras, inversions = parse_prompt(params)
|
prompt_pairs, loras, inversions = parse_prompt(params)
|
||||||
|
|
||||||
latents = get_latents_from_seed(params.seed, size)
|
latents = get_latents_from_seed(params.seed, size, params.batch)
|
||||||
pipe_type = params.get_valid_pipeline("txt2img")
|
pipe_type = params.get_valid_pipeline("txt2img")
|
||||||
pipe = load_pipeline(
|
pipe = load_pipeline(
|
||||||
server,
|
server,
|
||||||
|
|
Loading…
Reference in New Issue