Spaces:
Running
on
Zero
Running
on
Zero
Update autoregressive/models/generate.py
Browse files
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)
|