DataPilot
/

ArrowSmart_1.7b_instruction

Text Generation

text-generation-inference

Model card Files Files and versions

DataPilot commited on Apr 5, 2024

Commit

c488954

·

verified ·

1 Parent(s): 62558db

a

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ import torch
 from transformers import AutoModelForSequenceClassification
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline #transformerとtorchがインストールされていることを前提とします。
-model = AutoModelForCausalLM.from_pretrainedmodel = AutoModelForCausalLM.from_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
 tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
 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_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
 tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)