NNEngine
/

TinyWay-1.1.0

Text Generation

custom-architecture

Model card Files Files and versions

NNEngine commited on Jan 1

Commit

d91a44d

·

verified ·

1 Parent(s): 4551395

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -78,7 +78,7 @@ mdl = AutoModelForCausalLM.from_pretrained(model_id, config=config, trust_remote
 out = mdl.generate(
     **tok(
-        "It was scared to be more of the window and dad",
         return_tensors="pt"
     ).to(mdl.device),

 out = mdl.generate(
     **tok(
+        "Once upon a time",
         return_tensors="pt"
     ).to(mdl.device),