AI-Sweden-Models
/

tyr

Text Generation

text-generation-inference

Model card Files Files and versions

timpal0l commited on Mar 21, 2024

Commit

2ce3d85

·

verified ·

1 Parent(s): b5b2295

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -17,8 +17,11 @@ pipeline_tag: text-generation
 # How to use:
 ```python
 from transformers import pipeline
-pipe = pipeline("text-generation", "timpal0l/tyr", device="cuda:0")
 query = "Vad är grundlagen?"

 # How to use:
 ```python
+import torch
 from transformers import pipeline
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipeline("text-generation", "timpal0l/tyr", device=device)
 query = "Vad är grundlagen?"