Views
No views yet
| Name | Quant method | Size |
|---|---|---|
| Astrid-7B-LLama-Med.Q2_K.gguf | Q2_K | 2.36GB |
| Astrid-7B-LLama-Med.IQ3_XS.gguf | IQ3_XS | 2.6GB |
| Astrid-7B-LLama-Med.IQ3_S.gguf | IQ3_S | 2.75GB |
| Astrid-7B-LLama-Med.Q3_K_S.gguf | Q3_K_S | 2.75GB |
| Astrid-7B-LLama-Med.IQ3_M.gguf | IQ3_M | 2.9GB |
| Astrid-7B-LLama-Med.Q3_K.gguf | Q3_K | 3.07GB |
| Astrid-7B-LLama-Med.Q3_K_M.gguf | Q3_K_M | 3.07GB |
| Astrid-7B-LLama-Med.Q3_K_L.gguf | Q3_K_L | 3.35GB |
| Astrid-7B-LLama-Med.IQ4_XS.gguf | IQ4_XS | 3.4GB |
| Astrid-7B-LLama-Med.Q4_0.gguf | Q4_0 | 3.56GB |
| Astrid-7B-LLama-Med.IQ4_NL.gguf | IQ4_NL | 3.58GB |
| Astrid-7B-LLama-Med.Q4_K_S.gguf | Q4_K_S | 3.59GB |
| Astrid-7B-LLama-Med.Q4_K.gguf | Q4_K | 3.8GB |
| Astrid-7B-LLama-Med.Q4_K_M.gguf | Q4_K_M | 3.8GB |
| Astrid-7B-LLama-Med.Q4_1.gguf | Q4_1 | 3.95GB |
| Astrid-7B-LLama-Med.Q5_0.gguf | Q5_0 | 4.33GB |
| Astrid-7B-LLama-Med.Q5_K_S.gguf | Q5_K_S | 4.33GB |
| Astrid-7B-LLama-Med.Q5_K.gguf | Q5_K | 4.45GB |
| Astrid-7B-LLama-Med.Q5_K_M.gguf | Q5_K_M | 4.45GB |
| Astrid-7B-LLama-Med.Q5_1.gguf | Q5_1 | 4.72GB |
| Astrid-7B-LLama-Med.Q6_K.gguf | Q6_K | 5.15GB |
| Astrid-7B-LLama-Med.Q8_0.gguf | Q8_0 | 6.67GB |
transformers library on a machine with GPUs, first make sure you have the transformers library installed.pip install transformers==4.34.0token=True in the pipeline and login to hugginface_hub by running
python import huggingface_hub huggingface_hub.login(<ACCES_TOKEN>)
- Or directly pass your <ACCES_TOKEN> to token in the pipeline1from transformers import pipeline
2
3generate_text = pipeline(
4 model="PAIXAI/Astrid-7B-Med",
5 torch_dtype="auto",sssssssssssssssssssssssssssssssssssssssssssss
6 trust_remote_code=True,
7 use_fast=True,
8 device_map={"": "cuda:0"},
9 token=True,
10)
11
12res = generate_text(
13 "Why is drinking water so healthy?",
14 min_new_tokens=2,
15 max_new_tokens=256,
16 do_sample=False,
17 num_beams=1,
18 temperature=float(0.3),
19 repetition_penalty=float(1.2),
20 renormalize_logits=True
21)
22print(res[0]["generated_text"])print(generate_text.preprocess("Why is drinking water so healthy?")["prompt_text"])<|prompt|>Why is drinking water so healthy?</s><|answer|>transformers package, this will allow you to set trust_remote_code=False.1from h2oai_pipeline import H2OTextGenerationPipeline
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained(
5 "PAIXAI/Astrid-7B-Med",
6 use_fast=True,
7 padding_side="left",
8 trust_remote_code=True,
9)
10model = AutoModelForCausalLM.from_pretrained(
11 "PAIXAI/Astrid-7B-Med",
12 torch_dtype="auto",
13 device_map={"": "cuda:0"},
14 trust_remote_code=True,
15)
16generate_text = H2OTextGenerationPipeline(model=model, tokenizer=tokenizer)
17
18res = generate_text(
19 "Why is drinking water so healthy?",
20 min_new_tokens=2,
21 max_new_tokens=256,
22 do_sample=False,
23 num_beams=1,
24 temperature=float(0.3),
25 repetition_penalty=float(1.2),
26 renormalize_logits=True
27)
28print(res[0]["generated_text"])1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "PAIXAI/Astrid-7B-Med" # either local folder or huggingface model name
4# Important: The prompt needs to be in the same format the model was trained with.
5# You can find an example prompt in the experiment logs.
6prompt = "<|prompt|>How are you?</s><|answer|>"
7
8tokenizer = AutoTokenizer.from_pretrained(
9 model_name,
10 use_fast=True,
11 trust_remote_code=True,
12)
13model = AutoModelForCausalLM.from_pretrained(
14 model_name,
15 torch_dtype="auto",
16 device_map={"": "cuda:0"},
17 trust_remote_code=True,
18)
19model.cuda().eval()
20inputs = tokenizer(prompt, return_tensors="pt", add_special_tokens=False).to("cuda")
21
22# generate configuration can be modified to your needs
23tokens = model.generate(
24 input_ids=inputs["input_ids"],
25 attention_mask=inputs["attention_mask"],
26 min_new_tokens=2,
27 max_new_tokens=256,
28 do_sample=False,
29 num_beams=1,
30 temperature=float(0.3),
31 repetition_penalty=float(1.2),
32 renormalize_logits=True
33)[0]
34
35tokens = tokens[inputs["input_ids"].shape[1]:]
36answer = tokenizer.decode(tokens, skip_special_tokens=True)
37print(answer)load_in_8bit=True or load_in_4bit=True. Also, sharding on multiple GPUs is possible by setting device_map=auto.LlamaForCausalLM(
(model): LlamaModel(
(embed_tokens): Embedding(32000, 4096, padding_idx=0)
(layers): ModuleList(
(0-31): 32 x LlamaDecoderLayer(
(self_attn): LlamaAttention(
(q_proj): Linear(in_features=4096, out_features=4096, bias=False)
(k_proj): Linear(in_features=4096, out_features=4096, bias=False)
(v_proj): Linear(in_features=4096, out_features=4096, bias=False)
(o_proj): Linear(in_features=4096, out_features=4096, bias=False)
(rotary_emb): LlamaRotaryEmbedding()
)
(mlp): LlamaMLP(
(gate_proj): Linear(in_features=4096, out_features=11008, bias=False)
(up_proj): Linear(in_features=4096, out_features=11008, bias=False)
(down_proj): Linear(in_features=11008, out_features=4096, bias=False)
(act_fn): SiLUActivation()
)
(input_layernorm): LlamaRMSNorm()
(post_attention_layernorm): LlamaRMSNorm()
)
)
(norm): LlamaRMSNorm()
)
(lm_head): Linear(in_features=4096, out_features=32000, bias=False)
)