From 50a473c1b7ae72d0788e986efd67c541a179822d Mon Sep 17 00:00:00 2001 From: nullquant Date: Wed, 19 Jun 2024 19:23:58 +0300 Subject: [PATCH] external_embedding device to embedding device --- brushnet/powerpaint_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/brushnet/powerpaint_utils.py b/brushnet/powerpaint_utils.py index bcbb1f7..9bb84a6 100644 --- a/brushnet/powerpaint_utils.py +++ b/brushnet/powerpaint_utils.py @@ -375,7 +375,7 @@ class EmbeddingLayerWithFixes(nn.Module): start = external_embedding["start"] end = external_embedding["end"] target_ids_to_replace = [i for i in range(start, end)] - ext_emb = external_embedding["embedding"] + ext_emb = external_embedding["embedding"].to(embedding.device) # do not need to replace if not (input_ids == start).any():