nicolay-r commited on
Commit
65cabea
·
verified ·
1 Parent(s): 7139804

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-xl"
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