Arjun Bijanki

I'd also appreciate a license. Thanks for considering!

I found that on my GPU (NVidia T1200 Laptop GPU), the autocast to fp16 was causing the model to return garbage. Consider disabling it at this line: `torch.autocast(device_type='cuda', dtype=torch.float16)`