update

klonggan · Aug 30, 2023 · df0b748 · df0b748
1 parent 874737a
commit df0b748
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/llm_blender/blender/blender.py b/llm_blender/blender/blender.py
@@ -141,10 +141,9 @@ def fuse(
             logging.warning("No fuser loaded, please load fuser first through load_fuser()")
             return None
         generate_kwargs = generate_kwargs.copy()
-        max_length = generate_kwargs.pop("source_max_length", None) or self.fuser_config.max_length
         candidate_maxlength = generate_kwargs.pop("candidate_max_length", None) or self.fuser_config.candidate_maxlength
         dataset = GenFuserDataset(inputs, candidates, self.fuser_tokenizer,
-            instructions=instructions, max_length=max_length, 
+            instructions=instructions, max_length=self.fuser_config.max_length, 
             candidate_maxlength=candidate_maxlength)
 
         dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=False)