issues
search
ayaka14732
/
llama-2-jax
JAX implementation of the Llama 2 model
https://arxiv.org/abs/2307.09288
Creative Commons Zero v1.0 Universal
208
stars
23
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Generated tokens contain gibberish when giving multiple inputs, i.e. batch > 1
#29
ksmdnl
opened
5 months ago
0
Llama2 fine-tuned chat version
#28
ksmdnl
opened
5 months ago
0
Training From Scrach -- Redpajama
#27
opooladz
opened
7 months ago
0
Unable to shard llama 13B (and 70B) on v4-32 TPU
#26
defdet
opened
8 months ago
0
Allow for transfer learning
#25
defdet
closed
8 months ago
0
Update
#24
divyapatel4
closed
9 months ago
0
correct the formula for k
#23
defdet
closed
10 months ago
0
Problems sharding Llama-70B on TPU v3-32
#22
divyapatel4
opened
10 months ago
1
Adapt the repo for Mistral. Also error in calculation of `d_k` in `determine_params.py`
#21
defdet
closed
10 months ago
6
DEADLINE_EXCEEDED when running train.py on GPU node
#20
zigzagcai
opened
11 months ago
0
Improve generation speed and add benchmark for generation
#18
ayaka14732
opened
1 year ago
0
Fix training
#17
ayaka14732
closed
1 year ago
0
forward_llama() missing 1 required keyword-only argument: 'rotary_values'
#16
GluckLee
closed
1 year ago
2
Generation speed
#15
sh0416
opened
1 year ago
10
Got jax.errors.TracerIntegerConversionError when running generate.py
#14
zhangzx-uiuc
opened
1 year ago
2
Implement left padding
#13
ayaka14732
closed
1 year ago
0
Implement KV cache
#12
ayaka14732
closed
1 year ago
0
Update
#11
ayaka14732
closed
1 year ago
0
train.py OOM on TPUv3-8
#10
ethanhe42
opened
1 year ago
9
HF LLaMA Flax
#9
sanchit-gandhi
opened
1 year ago
1
13B parameter model
#7
aniquetahir
opened
1 year ago
0
Update
#6
ayaka14732
closed
1 year ago
0
Convert back to Hugging Face model
#5
ayaka14732
closed
1 year ago
0
Multihost training support
#4
ayaka14732
closed
1 year ago
0
Update
#3
ayaka14732
closed
1 year ago
0
Update to Llama 2
#2
ayaka14732
closed
1 year ago
0
Update to Llama 2
#1
ayaka14732
closed
1 year ago
0