DataPilot
/

ArrowSmartPlus_3.6B_instant_sft

Text Generation

text-generation-inference

Model card Files Files and versions Community

DataPilot commited on Mar 23

Commit

3b3e217

•

1 Parent(s): 78bcaad

aa

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -15,8 +15,8 @@ import torch
 from transformers import AutoModelForSequenceClassification
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline #transformerとtorchがインストールされていることを前提とします。
-model = AutoModelForCausalLM.from_pretrainedmodel = AutoModelForCausalLM.from_pretrained("DataPilot/ArrowSmartPlus_3.6B_instant_sft_JSHVer")
-tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmartPlus_3.6B_instant_sft_JSHVer")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)

 from transformers import AutoModelForSequenceClassification
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline #transformerとtorchがインストールされていることを前提とします。
+model = AutoModelForCausalLM.from_pretrainedmodel = AutoModelForCausalLM.from_pretrained("DataPilot/ArrowSmartPlus_3.6B_instant_sft")
+tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmartPlus_3.6B_instant_sft")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)