new model
This commit is contained in:
parent
38f3c0703f
commit
d3417ec437
1 changed files with 1 additions and 1 deletions
|
@ -22,7 +22,7 @@ import torch
|
|||
import asyncio
|
||||
|
||||
tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-large')
|
||||
model = AutoModelForCausalLM.from_pretrained('../southpark/output-small')
|
||||
model = AutoModelForCausalLM.from_pretrained('../southpark/output-medium')
|
||||
|
||||
# Twitter
|
||||
import tweepy
|
||||
|
|
Loading…
Add table
Reference in a new issue