Jaedeok Kim

Hi bharatv007 I guess you're thinking the first token (at step 0th) should be ignored in the computation of the log probability of contexts and the token ids should be...

Thanks for addressing the pynvml issue, relating to a driver version. @CoderHam can I know which doc(or link) you referred to determine the driver version (526)?