Model generated using the Glaive platform - https://app.glaive.ai

Use the following script-

from transformers import AutoModelForCausalLM, AutoTokenizer

import torch

model_path = "sahil2801/persuasion_model_glaive"

tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path).half().cuda()
prompt = '''Claim: <> \n\n Thought: <>'''

inputs = tokenizer(prompt,return_tensors="pt").to(model.device)
outputs = model.generate(**inputs,do_sample=True,eos_token_id=tokenizer.eos_token_id,temperature=0.9,max_new_tokens=1500)

print(tokenizer.decode(outputs[0],skip_special_tokens=True))
Downloads last month
12
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.