TeraSpace
/

dialofred

Text Generation

text2text-generation

text-generation-inference

Model card Files Files and versions

Update README.md

#1

by Den4ikAI - opened May 30, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -11,15 +11,15 @@ tags:
 - conversational
 ---
-How to use:
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 device='cuda'
-tokenizer = AutoTokenizer.from_pretrained('TeraSpace/dialofrednocontext_V2')
-model = AutoModelForSeq2SeqLM.from_pretrained('TeraSpace/dialofrednocontext_V2').to(device)
 while True:
     text_inp = input("=>")
     lm_text=f'<SC1>- {text_inp}\n- <extra_id_0>'

 - conversational
 ---
+# Usage
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 device='cuda'
+tokenizer = AutoTokenizer.from_pretrained('TeraSpace/dialofred')
+model = AutoModelForSeq2SeqLM.from_pretrained('TeraSpace/dialofred').to(device)
 while True:
     text_inp = input("=>")
     lm_text=f'<SC1>- {text_inp}\n- <extra_id_0>'