ibm-granite
/

granite-3.0-2b-base

Text Generation

Model card Files Files and versions Community

amezasor commited on Oct 20

Commit

b047a23

•

1 Parent(s): 96489ba

prompt update

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -242,7 +242,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