ToddGoldfarb commited on
Commit
2bb3bfd
1 Parent(s): e153f64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -37,9 +37,9 @@ cf.use_true_colors()
37
  cf.use_style('monokai')
38
  class CadetTinyAgent:
39
  def __init__(self):
40
- print(cf.bold | cf.purple("Waking up PIP..."))
41
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
42
- self.tokenizer = AutoTokenizer.from_pretrained("ToddGoldfarb/Cadet-Tiny", model_max_length=512)
43
  self.model = AutoModelForSeq2SeqLM.from_pretrained("ToddGoldfarb/Cadet-Tiny", low_cpu_mem_usage=True).to(self.device)
44
  self.conversation_history = ""
45
 
 
37
  cf.use_style('monokai')
38
  class CadetTinyAgent:
39
  def __init__(self):
40
+ print(cf.bold | cf.purple("Waking up Cadet-Tiny..."))
41
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
42
+ self.tokenizer = AutoTokenizer.from_pretrained("t5-small", model_max_length=512)
43
  self.model = AutoModelForSeq2SeqLM.from_pretrained("ToddGoldfarb/Cadet-Tiny", low_cpu_mem_usage=True).to(self.device)
44
  self.conversation_history = ""
45