DataPilot
/

ArrowSmartPlus_3.6B_instant_sft_JHSVer

Text Generation

text-generation-inference

Model card Files Files and versions

DataPilot commited on Mar 22, 2024

Commit

e0bb3da

·

verified ·

1 Parent(s): e475fe5

aasasdd

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_JHSVer")
+tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmartPlus_3.6B_instant_sft_JHSVer")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)