wondervictor commited on
Commit
80ddc5e
·
verified ·
1 Parent(s): 827e239

Update autoregressive/models/generate.py

Browse files
Files changed (1) hide show
  1. autoregressive/models/generate.py +1 -0
autoregressive/models/generate.py CHANGED
@@ -140,6 +140,7 @@ def generate(model, cond, max_new_tokens, emb_masks=None, cfg_scale=1.0, cfg_int
140
  condition = condition.to(torch.float32)
141
  print(condition)
142
  if condition is not None:
 
143
  condition = model.adapter(condition)
144
  condition = model.adapter_mlp(condition)
145
  print(condition)
 
140
  condition = condition.to(torch.float32)
141
  print(condition)
142
  if condition is not None:
143
+ print(model.adapter.model.embeddings.patch_embeddings.projection.weight)
144
  condition = model.adapter(condition)
145
  condition = model.adapter_mlp(condition)
146
  print(condition)