From 86b233a24b7981af5ef9e23df3f213557e928513 Mon Sep 17 00:00:00 2001 From: lin onetwo Date: Wed, 8 May 2024 22:07:54 +0800 Subject: [PATCH] fix: Add `.long()` to seq_orig --- rfdiffusion/inference/model_runners.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rfdiffusion/inference/model_runners.py b/rfdiffusion/inference/model_runners.py index 9a33d98..ec3b27b 100644 --- a/rfdiffusion/inference/model_runners.py +++ b/rfdiffusion/inference/model_runners.py @@ -304,7 +304,7 @@ def sample_init(self, return_forward_trajectory=False): xyz_27 = self.target_feats['xyz_27'] mask_27 = self.target_feats['mask_27'] - seq_orig = self.target_feats['seq'] + seq_orig = self.target_feats['seq'].long() L_mapped = len(self.contig_map.ref) contig_map=self.contig_map