Views
No views yet
h2ogpt-research-oasst1-llama-65b is a 65 billion parameter instruction-following large language model (NOT licensed for commercial use).
transformers library on a machine with GPUs, first make sure you have the following libraries installed.1pip install transformers==4.29.2
2pip install accelerate==0.19.0
3pip install torch==2.0.1
4pip install einops==0.6.11import torch
2from transformers import pipeline, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("h2oai/h2ogpt-research-oasst1-llama-65b", padding_side="left")
5generate_text = pipeline(model="h2oai/h2ogpt-research-oasst1-llama-65b", tokenizer=tokenizer, torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto", prompt_type="human_bot")
6res = generate_text("Why is drinking water so healthy?", max_new_tokens=100)
7print(res[0]["generated_text"])trust_remote_code=True you can download instruct_pipeline.py,
store it alongside your notebook, and construct the pipeline yourself from the loaded model and tokenizer:1import torch
2from h2oai_pipeline import H2OTextGenerationPipeline
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5tokenizer = AutoTokenizer.from_pretrained("h2oai/h2ogpt-research-oasst1-llama-65b", padding_side="left")
6model = AutoModelForCausalLM.from_pretrained("h2oai/h2ogpt-research-oasst1-llama-65b", torch_dtype=torch.bfloat16, device_map="auto")
7generate_text = H2OTextGenerationPipeline(model=model, tokenizer=tokenizer, prompt_type="human_bot")
8
9res = generate_text("Why is drinking water so healthy?", max_new_tokens=100)
10print(res[0]["generated_text"])LlamaForCausalLM(
(model): LlamaModel(
(embed_tokens): Embedding(32000, 8192, padding_idx=31999)
(layers): ModuleList(
(0-79): 80 x LlamaDecoderLayer(
(self_attn): LlamaAttention(
(q_proj): Linear(in_features=8192, out_features=8192, bias=False)
(k_proj): Linear(in_features=8192, out_features=8192, bias=False)
(v_proj): Linear(in_features=8192, out_features=8192, bias=False)
(o_proj): Linear(in_features=8192, out_features=8192, bias=False)
(rotary_emb): LlamaRotaryEmbedding()
)
(mlp): LlamaMLP(
(gate_proj): Linear(in_features=8192, out_features=22016, bias=False)
(down_proj): Linear(in_features=22016, out_features=8192, bias=False)
(up_proj): Linear(in_features=8192, out_features=22016, bias=False)
(act_fn): SiLUActivation()
)
(input_layernorm): LlamaRMSNorm()
(post_attention_layernorm): LlamaRMSNorm()
)
)
(norm): LlamaRMSNorm()
)
(lm_head): Linear(in_features=8192, out_features=32000, bias=False)
)1LlamaConfig {
2 "_name_or_path": "h2oai/h2ogpt-research-oasst1-llama-65b",
3 "architectures": [
4 "LlamaForCausalLM"
5 ],
6 "bos_token_id": 0,
7 "custom_pipelines": {
8 "text-generation": {
9 "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
10 "pt": "AutoModelForCausalLM"
11 }
12 },
13 "eos_token_id": 1,
14 "hidden_act": "silu",
15 "hidden_size": 8192,
16 "initializer_range": 0.02,
17 "intermediate_size": 22016,
18 "max_position_embeddings": 2048,
19 "max_sequence_length": 2048,
20 "model_type": "llama",
21 "num_attention_heads": 64,
22 "num_hidden_layers": 80,
23 "pad_token_id": -1,
24 "rms_norm_eps": 1e-05,
25 "tie_word_embeddings": false,
26 "torch_dtype": "float16",
27 "transformers_version": "4.30.1",
28 "use_cache": true,
29 "vocab_size": 32000
30}
31