Fix attention mask dtype issues.

amplify.py

@@ -248,7 +248,7 @@ class AMPLIFY(AMPLIFYPreTrainedModel):
         if attention_mask is not None and not torch.all(attention_mask == 0):
             assert attention_mask.dtype != torch.bool and 1.0 not in attention_mask, (
                 "AMPLIFY expects an additive attention_mask.\n"
-                "Modify the output of the tokenizer with attention_mask = torch.where(attention_mask, float(0.0), float("-inf"))"
+                "Modify the output of the tokenizer with attention_mask = torch.where(attention_mask, float(0.0), float('-inf'))"
             )
             attention_mask = (
                 attention_mask.unsqueeze(1)