Update app.py

app.py

@@ -47,7 +47,7 @@ class LocalBlend:
         mask = nnf.interpolate(mask, size=(x_t.shape[2:]))
         mask = mask / mask.max(2, keepdims=True)[0].max(3, keepdims=True)[0]
         mask = mask.gt(self.threshold)
-        mask = (mask[:1] + mask[1:]).float()
+        mask = (mask[:1] + mask[1:]).to(x_t.dtype)
         x_t = x_t[:1] + mask * (x_t - x_t[:1])
         return x_t
 
@@ -221,7 +221,7 @@ def get_equalizer(text: str, word_select: Union[int, Tuple[int, ...]], values: U
     if type(word_select) is int or type(word_select) is str:
         word_select = (word_select,)
     equalizer = torch.ones(len(values), 77)
-    values = torch.tensor(values, dtype=torch.float32)
+    values = torch.tensor(values, dtype=torch_dtype)
     for word in word_select:
         inds = ptp_utils.get_word_inds(text, word, tokenizer)
         equalizer[:, inds] = values