mrSoul7766
commited on
Commit
•
51cd6a3
1
Parent(s):
2bad44b
Update README.md
Browse files
README.md
CHANGED
@@ -52,8 +52,8 @@ I'm on it! I'm here to assist you in tracking the compensation. To track the com
|
|
52 |
# Load model directly
|
53 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
54 |
|
55 |
-
tokenizer = AutoTokenizer.from_pretrained("mrSoul7766/chat-
|
56 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("mrSoul7766/chat-
|
57 |
|
58 |
# Set maximum generation length
|
59 |
max_length = 512
|
|
|
52 |
# Load model directly
|
53 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
54 |
|
55 |
+
tokenizer = AutoTokenizer.from_pretrained("mrSoul7766/CUSsupport-chat-t5-small")
|
56 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("mrSoul7766/CUSsupport-chat-t5-small")
|
57 |
|
58 |
# Set maximum generation length
|
59 |
max_length = 512
|