ibm-ai-platform
/

Bamba-9B-v2

Text Generation

Model card Files Files and versions

divykum commited on Apr 25

Commit

2b4b4e8

·

verified ·

1 Parent(s): e25d607

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -65,8 +65,8 @@ You can utilize our newly contributed HF integration to run inference on our Bam
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("ibm-fms/Bamba-9B-v2")
-tokenizer = AutoTokenizer.from_pretrained("ibm-fms/Bamba-9B-v2")
 message = ["Mamba is a snake with following properties  "]
 inputs = tokenizer(message, return_tensors='pt', return_token_type_ids=False)

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("ibm-ai-platform/Bamba-9B-v2")
+tokenizer = AutoTokenizer.from_pretrained("ibm-ai-platform/Bamba-9B-v2")
 message = ["Mamba is a snake with following properties  "]
 inputs = tokenizer(message, return_tensors='pt', return_token_type_ids=False)