problems about the random seed in the code

[snowflakewang]

hello, thank you for your brilliant work in the fast feed-forward 3d generation model!
i have tried your HuggingFace demo and it works well. I noticed that users can manipulate the random seed in the demo. but when I run your code, it seems no command arguments for the random seed. i wonder how can I set the seed properly. np? torch? torch.cuda?

thank you :)

[thuwzy]

Hi! You can manually call the set_seed() function in the pipeline class. https://github.com/thu-ml/CRM/blob/3e677cb41d4856dc7c46c5dace00d52336f5c614/pipelines.py#L129-#L132

[snowflakewang]

Hi! You can manually call the set_seed() function in the pipeline class. https://github.com/thu-ml/CRM/blob/3e677cb41d4856dc7c46c5dace00d52336f5c614/pipelines.py#L129-#L132

Thank you for your reply! I have solved this problem.
Have a good day :)