CyberNativeAI commited on
Commit
6fa4e78
·
verified ·
1 Parent(s): 3a4e553

Align the inference example with the final DPO checkpoint

Browse files

The inference example on this final DPO model card assigns `allenai/AstaBrief_8B_SFT` to `model_name`. The tokenizer, Transformers model, and vLLM model loaders use that binding. If the example is intended for the final checkpoint described by this card, changing the assignment to `allenai/AstaBrief_8B` aligns those declared identifiers with the card.

Checked against [README revision 3a4e553](https://huggingface.co/allenai/AstaBrief_8B/blob/3a4e553a40c87d0f426c255267a367c7f721a153/README.md). I checked the syntax and identifiers without loading either checkpoint or running inference; no performance or model-quality conclusion follows. The SFT model card's existing assignment matches its own checkpoint.

Prepared with OpenAI Codex assistance for CyberNative AI LLC. Correction contact: hello@cybernative.ai.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ Following is an example inference code snippet:
29
  from transformers import AutoModelForCausalLM, AutoTokenizer
30
  from vllm import LLM, SamplingParams
31
 
32
- model_name = "allenai/AstaBrief_8B_SFT"
33
 
34
  # load the tokenizer and the model
35
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
29
  from transformers import AutoModelForCausalLM, AutoTokenizer
30
  from vllm import LLM, SamplingParams
31
 
32
+ model_name = "allenai/AstaBrief_8B"
33
 
34
  # load the tokenizer and the model
35
  tokenizer = AutoTokenizer.from_pretrained(model_name)