NickyNicky
/

gemma-2b-it_oasst2_chatML_Cluster2_aya_multilingual_10k_prompts_ranked_all_json_V1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

NickyNicky commited on Mar 13

Commit

271c873

•

1 Parent(s): d27316a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -101,8 +101,8 @@ from transformers import StoppingCriteria, StoppingCriteriaList
 import torch
-model_id='NickyNicky/gemma-2b-it_oasst2_chatML_Cluster2_aya_multilingual'
 model = AutoModelForCausalLM.from_pretrained(model_id,
                                              device_map="auto",
                                              trust_remote_code=True,

 import torch
+# model_id='NickyNicky/gemma-2b-it_oasst2_chatML_Cluster2_aya_multilingual'
+model_id= "NickyNicky/gemma-2b-it_oasst2_chatML_Cluster2_aya_multilingual_10k_prompts_ranked_all_json_V1"
 model = AutoModelForCausalLM.from_pretrained(model_id,
                                              device_map="auto",
                                              trust_remote_code=True,