hishab
/

titulm-mpt-1b-v1.0

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sagorsarker commited on Apr 3

Commit

a4f214e

•

1 Parent(s): 7fc2b28

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ model = transformers.AutoModelForCausalLM.from_pretrained(
 tokenizer = transformers.AutoTokenizer.from_pretrained('hishab/titulm-1b-bn-v1')
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')
-output = pipe('আমি বাংলায় গান\n',
             max_new_tokens=100,
             do_sample=True,
             use_cache=True)

 tokenizer = transformers.AutoTokenizer.from_pretrained('hishab/titulm-1b-bn-v1')
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')
+output = pipe('আমি বাংলায় গান',
             max_new_tokens=100,
             do_sample=True,
             use_cache=True)