Kushagra_Bhatia:

Hq = self.lq(hq)


@Kushagra_Bhatia Another keen look at the traceback and another thought. It’s evident the issue is at the  line. How about sending the device argument to the model and then putting hq and later hc on the device? I guess I was little sloppy earlier.