Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ import torch
|
|
44 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
45 |
|
46 |
# Setup model path.
|
47 |
-
model_path = "nicolay-r/flan-t5-tsa-thor-
|
48 |
# Setup device.
|
49 |
device = "cuda:0"
|
50 |
|
|
|
44 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
45 |
|
46 |
# Setup model path.
|
47 |
+
model_path = "nicolay-r/flan-t5-tsa-thor-base"
|
48 |
# Setup device.
|
49 |
device = "cuda:0"
|
50 |
|