Hugging Face's logo

Spaces:
wondervictor
/
ControlAR
Running on Zero

App Files Community
1
wondervictor commited on
Commit
5559068
·
verified ·
1 Parent(s): b8ed191

Update autoregressive/models/generate.py

Browse files
Files changed (1) hide show
  1. autoregressive/models/generate.py +2 -1
autoregressive/models/generate.py CHANGED
@@ -139,7 +139,8 @@ def decode_n_tokens(
139
  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):
140
  if condition is not None:
141
  with torch.no_grad():
142
- print(f'nan: {torch.any(torch.isinf(model.adapter.model.embeddings.patch_embeddings.projection.weight))}')
 
143
  condition = model.adapter(condition)
144
  print(condition)
145
  condition = model.adapter_mlp(condition)
 
139
  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):
140
  if condition is not None:
141
  with torch.no_grad():
142
+ print(f'nan: {torch.any(torch.isnan(model.adapter.model.embeddings.patch_embeddings.projection.weight))}')
143
+ print(model.adapter.model.embeddings.patch_embeddings.projection.weight)
144
  condition = model.adapter(condition)
145
  print(condition)
146
  condition = model.adapter_mlp(condition)