Model Card for Model ID

Model Details

Model Description

Cicero italian legal assistant version 2. Based upon Phi1.5, fine-tuned with 1.5GB of italian civil judgements.

Developed by: Marco Calamo, Francesca De Luzi, Mattia Macrì, Tommaso Mencattini, Massimo Mecella
Model type: Phi1.5
Language (NLP): Italian
License: Apache 2.0

Model Sources

Repository: Github
Paper: Under review

Uses

Use Cicero to autocomplete sentences in a legal italian context.

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import  AutoModelForCausalLM, AutoTokenizer 
import torch

model =  AutoModelForCausalLM.from_pretrained("DIAG-PSSeng/cicero_v2-phi1.5", trust_remote_code=True ,torch_dtype=torch.float16).to("cuda")

tokenizer = AutoTokenizer.from_pretrained("DIAG-PSSeng/cicero_v2-phi1.5", trust_remote_code=True)

def generate_text(model, tokenizer, prompt, length=50, do_sample=True):
    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
    gen_tokens = model.generate(**inputs,do_sample=True,temperature=0.9, min_length=length,max_length=length)
    generated_text = tokenizer.batch_decode(gen_tokens)
    return generated_text

print(generate_text(model=model, tokenizer=tokenizer, prompt="Con ricorso del"))

Citation

BibTeX: Under review