AI-Sweden-Models
/

gpt-sw3-356m

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

JoeyOhman commited on Jan 10, 2023

Commit

9704f32

•

1 Parent(s): ab6dc43

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 # Initialize Variables
-model_name = "AI-Sweden/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"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 prompt = "Träd är fina för att"