nicholasKluge
/

Aira-2-124M-DPO

Text Generation

instruction tuned

text generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Dec 10, 2023

Commit

9d30d2b

•

1 Parent(s): f1b6115

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -77,8 +77,8 @@ import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/Aira-2-124M')
-aira = AutoModelForCausalLM.from_pretrained('nicholasKluge/Aira-2-124M')
 aira.eval()
 aira.to(device)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/Aira-2-124M-DPO')
+aira = AutoModelForCausalLM.from_pretrained('nicholasKluge/Aira-2-124M-DPO')
 aira.eval()
 aira.to(device)