Spaces:

callmeclover
/

Remi

Runtime error

Remi / app.py

Update app.py

7ccaa61 over 1 year ago

No virus

373 Bytes

	import gradio as gr
	from transformers import AutoTokenizer, AutoModelForCausalLM

	tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")

	model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")

	inputs = tokenizer("Hello", return_tensors="pt")
	outputs = model(**inputs, labels=inputs["input_ids"])
	loss = outputs.loss
	logits = outputs.logits