AI-Sweden-Models
/

gpt-sw3-1.3b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

JoeyOhman commited on Jan 10, 2023

Commit

b49413d

·

1 Parent(s): 8ca115f

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-1.3b"
 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-1.3b"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 prompt = "Träd är fina för att"