1 files changed, 1 insertions, 1 deletions
diff --git a/model.py b/model.py
index d3716d1..567e533 100644
--- a/model.py
+++ b/model.py
@@ -242,7 +242,7 @@ class Generator(Runner):
         super().__init__(model)
         self.cuda = cuda
 
-    def __call__(self, n_seqs, seq_len):
+    def __call__(self, n_seqs, seq_len, primer):
         # generation doesn't work with CUDNN for some reason
         torch.backends.cudnn.enabled = False