Updates t5 model to load with cpu.

model/t5.py

@@ -20,7 +20,7 @@ class T5ClassificationModel(nn.Module):
             self.base_model = T5EncoderModel(config)
 
             # Load the state_dict and remove unwanted keys
-            state_dict = torch.load(pytorch_model_path)
+            state_dict = torch.load(pytorch_model_path, map_location=torch.device("cpu"))
             filtered_state_dict = {
                 k.replace("base_model.", ""): v
                 for k, v in state_dict.items()