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

added notification

Browse source
This commit is contained in:
Tom Young 2024-06-29 06:03:48 +00:00
parent fdcd22257c
commit a2bc2fde96
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/caption/pllava_dir/caption_pllava.py
View file

@ -375,7 +375,7 @@ def main():
# write the dataframe to a new csv file called '*_pllava_13b_caption.csv'
new_csv_path = args.csv_path.replace(".csv", "_text.csv")
df.to_csv(new_csv_path, index=False)
print(f"Results saved to {new_csv_path}")
if __name__ == "__main__":
main()