Update autoregressive/models/generate.py

autoregressive/models/generate.py

@@ -171,7 +171,7 @@ def generate(model, cond, max_new_tokens, emb_masks=None, cfg_scale=1.0, cfg_int
             
             if condition is not None:
                 condition_null = torch.zeros_like(condition)
-                condition_combined = torch.cat((condition, condition_null), dim=0)
+                condition_combined = torch.cat((condition, condition), dim=0)
             else:
                 condition_combined = None
         else: