remove config.json dependency, default to torch in image_from_text.py

2022-07-01 12:03:37 -04:00
parent 4404e70764
commit 85f5866eff
10 changed files with 52 additions and 64 deletions
@@ -130,7 +130,6 @@ def keep_top_k(logits: jnp.ndarray, k: int) -> jnp.ndarray:

 class DalleBartDecoderFlax(nn.Module):
    image_token_count: int
-    text_token_count: int
    image_vocab_count: int
    attention_head_count: int
    embed_count: int