OpenDFM
/

RetroDFM-R-v0-8B

Text Generation

Model card Files Files and versions

Situo commited on 8 days ago

Commit

c828b32

·

verified ·

1 Parent(s): 989dae9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ import re
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
-model_name_or_id = "OpenDFM/RetroDFM-R-8B"
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_id)
 model = AutoModelForCausalLM.from_pretrained(model_name_or_id, torch_dtype=torch.bfloat16, device_map="auto")

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
+model_name_or_id = "OpenDFM/RetroDFM-R-v0-8B"
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_id)
 model = AutoModelForCausalLM.from_pretrained(model_name_or_id, torch_dtype=torch.bfloat16, device_map="auto")