Spaces:

saksornr
/

personaGPT_custom

Runtime error

Saksorn ruangtanusak

Duplicate from ngxson/poet-cat

e2c5599 11 months ago

No virus

345 Bytes

	from transformers import GPT2Tokenizer, GPT2LMHeadModel, AutoTokenizer, AutoModelForCausalLM
	import os
	import torch

	print("TRANSFORMERS_CACHE", os.environ['TRANSFORMERS_CACHE'])
	print("Fetching model...")

	tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
	model = AutoModelForCausalLM.from_pretrained("af1tang/personaGPT")