From 51e20ae01e81f60edf6c50565f9d09f126f9d4b3 Mon Sep 17 00:00:00 2001 From: Adam <24621027+adoyle0@users.noreply.github.com> Date: Thu, 9 Feb 2023 09:50:21 -0500 Subject: [PATCH] cartman moved --- src/cartman.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cartman.py b/src/cartman.py index 1f898ed..d97468f 100644 --- a/src/cartman.py +++ b/src/cartman.py @@ -12,7 +12,7 @@ from transformers.models.auto.modeling_auto import AutoModelForCausalLM from transformers.models.auto.tokenization_auto import AutoTokenizer 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): input_ids = tokenizer(input_text + tokenizer.eos_token, return_tensors="pt").input_ids