Llama 35d344b951 Remove torch dependency and more generic dim0 workaround ...

Remove torch dependency from hf.py
Make workaround for dimension zero values of token_ids
more generic to handle every token, not just newlines.

2023-05-03 09:48:16 -07:00

..

inference_models

Remove torch dependency and more generic dim0 workaround

2023-05-03 09:48:16 -07:00

inference_model.py

Merge pull request #342 from one-some/model-structure-and-maybe-rwkv

2023-05-03 03:34:17 +02:00

lazy_loader.py

Lazyload: Whoops

2023-04-25 18:46:54 -05:00

logits_processors.py

Modeling: Fix logits processors (probs, biasing, lua)

2023-03-17 16:56:47 -05:00

patches.py

Model: And another refactor

2023-03-01 19:16:35 -06:00

post_token_hooks.py

Model: Respect model lazyload over kaivars

2023-03-10 20:00:39 -06:00

stoppers.py

Potential fix

2023-04-27 19:51:10 -05:00

test_generation.py

Model: More Jax import fixes and formatting

2023-03-17 15:36:44 -05:00

tokenizer.py

Fix tokenizer fallback for llama

2023-05-01 19:42:52 -05:00

warpers.py

Undo pretty code because I haven't cracked the jax enigma yet

2023-04-25 21:11:49 -05:00