I-BRICKS
/

Cerebro_BM_solar_v01

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

spow12 commited on Jul 22

Commit

31bbd05

•

1 Parent(s): bdb6edb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ language:
 ```
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-model_id = "I-BRICKS/Cerebro_BM_solar"
 model = AutoModelForCausalLM.from_pretrained(
         model_id,
         torch_dtype=torch.float16,

 ```
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+model_id = "I-BRICKS/Cerebro_BM_solar_v01"
 model = AutoModelForCausalLM.from_pretrained(
         model_id,
         torch_dtype=torch.float16,