Commit
•
2bb3bfd
1
Parent(s):
e153f64
Update README.md
Browse files
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
|
41 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
42 |
-
self.tokenizer = AutoTokenizer.from_pretrained("
|
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 |
|