@viralpandey were you able to resolve your problem? I’m suffering from a similar issue where I can confirm that my model/guide have the same size of tensors, but I get the same error
ValueError: at site "obs", invalid log_prob shape
Expected [], actual [20]