Update autoregressive/models/generate.py

autoregressive/models/generate.py

@@ -209,6 +209,7 @@ def generate(model, cond, max_new_tokens, emb_masks=None, cfg_scale=1.0, cfg_int
     input_pos = torch.arange(0, T, device=device)
     next_token = prefill(model, cond_combined, input_pos, cfg_scale, condition_combined, control_strength,**sampling_kwargs)
     seq[:, T:T+1] = next_token
+    print(model.control_strength)
 
     input_pos = torch.tensor([T], device=device, dtype=torch.int)
     generated_tokens, _ = decode_n_tokens(model, next_token, input_pos, max_new_tokens-1, cfg_scale, cfg_interval, condition=condition_combined, **sampling_kwargs)