FatihJimale
/

gpt2-medium-somali

Model card Files Files and versions

AbdisamadO commited on Oct 25, 2025

Commit

d2ff7ba

·

verified ·

1 Parent(s): 26f2cca

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ model_id = "FatihJimale/gpt2-medium-somali"
 tok = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
-prompt = "qarax xoogan ayaa ka dhacay magaalada"
 inputs = tok(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
     **inputs,
@@ -107,7 +107,7 @@ model_id = "FatihJimale/gpt2-medium-somali"
 tok = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
-prompt = "qarax xoogan ayaa ka dhacay magaalada"
 inputs = tok(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=80, do_sample=True, temperature=0.9, top_p=0.92, repetition_penalty=1.08)
 print(tok.decode(outputs[0], skip_special_tokens=True))

 tok = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
+prompt = "roob xoogan ayaa ka da'ay magaalada"
 inputs = tok(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
     **inputs,
 tok = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
+prompt = "roob xoogan ayaa ka da'ay magaalada"
 inputs = tok(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=80, do_sample=True, temperature=0.9, top_p=0.92, repetition_penalty=1.08)
 print(tok.decode(outputs[0], skip_special_tokens=True))