Select Git revision
shape-transformer-rwkv
-
-
- Open in your IDE
- Download source code
Forked from
Gregor Kobsik / Shape Transformer
23 commits behind the upstream repository.

Gregor Kobsik
authored
- add API for token_embedding, transformer_module and generative_head to architecture implementation - extend composite head to process only last depth layer - extend ShapeSampler to accept "recurrent" implementation. TODO: set recurrent sampler for "fast" architecture as default. - implemented recurrent sampler + basic/substitution/composite token generator
Name | Last commit | Last update |
---|---|---|
callbacks | ||
configs | ||
data | ||
evaluation | ||
executable | ||
modules | ||
sample | ||
tests | ||
utils | ||
.gitignore | ||
main.py | ||
parameter_search.py | ||
requirements.txt | ||
sample.py | ||
sample_conditional.py | ||
setup.cfg |