adam/lulzbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

cartman moved

Browse source
This commit is contained in:
Adam 2023-02-09 09:50:21 -05:00
parent faafd258a3
commit 51e20ae01e
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/cartman.py
View file

@ -12,7 +12,7 @@ from transformers.models.auto.modeling_auto import AutoModelForCausalLM
from transformers.models.auto.tokenization_auto import AutoTokenizer from transformers.models.auto.tokenization_auto import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large") tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
model = AutoModelForCausalLM.from_pretrained("../chatbots/southpark/cartman/models/output-medium") model = AutoModelForCausalLM.from_pretrained("../cartman/train/cartman/models/output-medium")
def cartman_speak(input_text): def cartman_speak(input_text):
input_ids = tokenizer(input_text + tokenizer.eos_token, return_tensors="pt").input_ids input_ids = tokenizer(input_text + tokenizer.eos_token, return_tensors="pt").input_ids