ibm-granite
/

granite-3.0-1b-a400m-base

Text Generation

Model card Files Files and versions Community

amezasor commited on 18 days ago

Commit

764f692

•

1 Parent(s): 82ba732

prompt update

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -243,7 +243,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
 model.eval()
 # change input text as desired
-input_text = "Where is the MIT-IBM Watson AI Lab located?"
 # tokenize the text
 input_tokens = tokenizer(input_text, return_tensors="pt").to(device)
 # generate output tokens

 model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
 model.eval()
 # change input text as desired
+input_text = "Where is the Thomas J. Watson Research Center located?"
 # tokenize the text
 input_tokens = tokenizer(input_text, return_tensors="pt").to(device)
 # generate output tokens