Hugging Face's logo

MaxJeblick
/
llama2-0b-unit-test

Text Generation
Transformers
PyTorch
Safetensors
llama
text-generation-inference
Model card Files
xet
Community
1
MaxJeblick commited on
Commit
ba8a171
·
verified ·
1 Parent(s): 7d94b41

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ repo_name = "MaxJeblick/llama2-0b-unit-test"
11
  model_name = "h2oai/h2ogpt-4096-llama2-7b-chat"
12
  config = AutoConfig.from_pretrained(model_name)
13
  config.hidden_size = 12
14
- config.max_position_embeddings = 32
15
  config.intermediate_size = 24
16
  config.num_attention_heads = 2
17
  config.num_hidden_layers = 2
 
11
  model_name = "h2oai/h2ogpt-4096-llama2-7b-chat"
12
  config = AutoConfig.from_pretrained(model_name)
13
  config.hidden_size = 12
14
+ config.max_position_embeddings = 1024
15
  config.intermediate_size = 24
16
  config.num_attention_heads = 2
17
  config.num_hidden_layers = 2