Views
No views yet
talkie-lm/talkie-web-13b-base, the original Talkie base completion model.talkie-lm/talkie-1930-13b-base and intended for controlled comparisons between vintage and modern language models.trust_remote_code modules and BF16 sharded safetensors.trust_remote_code=TrueAutoTokenizer1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4path = "xlr8harder/talkie-web-13b-base-tf"
5tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 path,
8 trust_remote_code=True,
9 dtype=torch.bfloat16,
10 device_map={"": "cuda"},
11 use_safetensors=True,
12)1inputs = tokenizer("The latest discoveries in physics suggest that", return_tensors="pt").to("cuda")
2output = model.generate(**inputs, max_new_tokens=64)
3print(tokenizer.decode(output[0], skip_special_tokens=True))lm_head_gain is folded into
lm_head.weight during conversion; the other Talkie gain parameters remain
explicit model parameters. Using vLLM's logit_scale-style approach was not
used because it applies scaling after the output matmul, while Talkie applies
the gain to the head weight before the matmul. In BF16 this can introduce small
rounding differences and, in smoke tests, changed one near-tied top-token
ordering.1vllm serve xlr8harder/talkie-web-13b-base-tf \
2 --task generate \
3 --model-impl transformers \
4 --trust-remote-code \
5 --dtype bfloat16 \
6 --max-model-len 40960.03125.