nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge branch 'dev/v1.2' into vbench_trials

Browse source
This commit is contained in:
Shen-Chenhui 2024-06-17 03:13:52 +00:00
commit 32d10d4796
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
scripts/inference.py
View file

@ -149,6 +149,7 @@ def main():
# == get json from prompts ==
batch_prompts, refs, ms = extract_json_from_prompts(batch_prompts, refs, ms)
original_batch_prompts = batch_prompts
# == get reference for condition ==
refs = collect_references_batch(refs, vae, image_size)
@ -166,7 +167,7 @@ def main():
save_dir,
sample_name=sample_name,
sample_idx=start_idx + idx,
prompt=batch_prompts[idx],
prompt=original_batch_prompts[idx],
prompt_as_path=prompt_as_path,
num_sample=num_sample,
k=k,