From 884563c0286bea1fb080fd3124687b344bb9c9f0 Mon Sep 17 00:00:00 2001 From: laksjdjf Date: Wed, 30 Apr 2025 23:52:28 +0900 Subject: [PATCH] =?UTF-8?q?clone=E5=BF=85=E8=A6=81=E3=81=8B=E3=82=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/reference/reference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/reference/reference.py b/scripts/reference/reference.py index 845c3c3..a052d08 100644 --- a/scripts/reference/reference.py +++ b/scripts/reference/reference.py @@ -60,8 +60,8 @@ class ReferenceApply: if end_sigma <= sigma <= start_sigma and block_number <= self.depth: - k_ref = k_out[index::batch_size].repeat_interleave(batch_size, dim=0) - v_ref = v_out[index::batch_size].repeat_interleave(batch_size, dim=0) + k_ref = k_out[index::batch_size].repeat_interleave(batch_size, dim=0).clone() + v_ref = v_out[index::batch_size].repeat_interleave(batch_size, dim=0).clone() k_out = torch.cat([k_out, k_ref], dim=1) if mode == "concat" else k_ref v_out = torch.cat([v_out, v_ref], dim=1) if mode == "concat" else v_ref