AI-Sweden-Models
/

gpt-sw3-356m-instruct

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

apsod commited on Apr 28, 2023

Commit

923366c

•

1 Parent(s): c3238f1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 # Initialize Variables
-model_name = "AI-Sweden-Models/gpt-sw3-356m"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 prompt = "Träd är fina för att"

 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 # Initialize Variables
+model_name = "AI-Sweden-Models/gpt-sw3-356m-instruct"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 prompt = "Träd är fina för att"