| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| model_path = '/home/energyxadmin/UI2/merge' | |
| model = AutoModelForCausalLM.from_pretrained(model_path) | |
| tokenizer = AutoTokenizer.from_pretrained(model_path) | |
| # Example text generation | |
| input_ids = tokenizer.encode("What song did Eric Pask write or was a part of", return_tensors="pt") | |
| generated_text_samples = model.generate(input_ids, max_length=1000) | |
| print("Generated text:", tokenizer.decode(generated_text_samples[0], skip_special_tokens=True)) | |