Closed hanruijiang closed 1 month ago
Hi @hanruijiang, I was able to replicate exactly as you described ... it seems like model explorer expects all arguments to be tensors: https://github.com/google-ai-edge/model-explorer/blob/main/src/server/package/src/model_explorer/pytorch_exported_program_adater_impl.py#L177
Hi @yijie-yang, can you please take a look?
Hi @hanruijiang, I was able to replicate exactly as you described ... it seems like model explorer expects all arguments to be tensors: https://github.com/google-ai-edge/model-explorer/blob/main/src/server/package/src/model_explorer/pytorch_exported_program_adater_impl.py#L177
Hi @yijie-yang, can you please take a look?
Hi @pkgoogle ,
You're right. I fixed this bug by remove _'returndict': False in the kwargs .
Thank you for your help.
@hanruijiang, Thanks, yeah that worked for me as well... if you have no more open items, feel free to close, thanks.
Hi,
I can't run model_explorer on UNet2DConditionModel from diffusers lib.
Could you please help me solve this problem?
Environment
google colab
Steps to Reproduce:
Error log