Update autoregressive/models/generate.py

autoregressive/models/generate.py

@@ -139,7 +139,8 @@ def decode_n_tokens(
 def generate(model, cond, max_new_tokens, emb_masks=None, cfg_scale=1.0, cfg_interval=-1, condition=None, condition_null=None, condition_token_nums=0, **sampling_kwargs):
     if condition is not None:
         with torch.no_grad():
-            print(f'nan: {torch.any(torch.isinf(model.adapter.model.embeddings.patch_embeddings.projection.weight))}')
+            print(f'nan: {torch.any(torch.isnan(model.adapter.model.embeddings.patch_embeddings.projection.weight))}')
+            print(model.adapter.model.embeddings.patch_embeddings.projection.weight)
             condition = model.adapter(condition)
             print(condition)
             condition = model.adapter_mlp(condition)