Update app.py

app.py

@@ -198,7 +198,7 @@ class AttentionReplace(AttentionControlEdit):
     def __init__(self, prompts, num_steps: int, cross_replace_steps: float, self_replace_steps: float,
                  local_blend: Optional[LocalBlend] = None):
         super(AttentionReplace, self).__init__(prompts, num_steps, cross_replace_steps, self_replace_steps, local_blend)
-        self.mapper = seq_aligner.get_replacement_mapper(prompts, tokenizer).to(device).to(torch_dtype)
+        self.mapper = seq_aligner.get_replacement_mapper(prompts, tokenizer).to(device)
 
 
 class AttentionRefine(AttentionControlEdit):
@@ -212,7 +212,7 @@ class AttentionRefine(AttentionControlEdit):
                  local_blend: Optional[LocalBlend] = None):
         super(AttentionRefine, self).__init__(prompts, num_steps, cross_replace_steps, self_replace_steps, local_blend)
         self.mapper, alphas = seq_aligner.get_refinement_mapper(prompts, tokenizer)
-        self.mapper, alphas = self.mapper.to(device).to(torch_dtype), alphas.to(device).to(torch_dtype)
+        self.mapper, alphas = self.mapper.to(device), alphas.to(device).to(torch_dtype)
         self.alphas = alphas.reshape(alphas.shape[0], 1, 1, alphas.shape[1])