---
license: mit
base_model:
- microsoft/phi-1_5
---
# Model Card for Model ID

### Summary

<!-- Provide a quick summary of what the model is/does. -->

This is a supervised fine-tuned model for text completion based on Phi 1.5. It has been finetuned on a filtered version of the The Complete Works of William Shakespeare, which can be found and downloaded from here: [https://www.gutenberg.org/ebooks/100](https://www.gutenberg.org/ebooks/100).

### Model Description

<!-- Provide a longer summary of what this model is. -->

- **Developed by:** Course Organizers
- **Finetuned from model:** microsoft/phi-1_5

### Training Details

This model has been trained using the TLR library and SFTTrainer class from Huggingface.

### Training Data

<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->

The Complete Works of William Shakespeare, which can be found and downloaded from here: [https://www.gutenberg.org/ebooks/100](https://www.gutenberg.org/ebooks/100)

#### Training Hyperparameters

The following hyperparameters were used during training:

 - learning_rate: 1e-05
 - per_device_train_batch_size: 4
 - seed: 42
 - gradient_accumulation_steps: 1
 - optimizer: Adam with betas=(0.9, 0.95) 
 - lr_scheduler_type: linear
 - weight_decay: 0.1
 - num_epochs: 1


### Framework Versions

 - accelerate==0.26.1
 - datasets==2.16.1
 - transformers==4.45.2
 - trl==0.11.2

### Compute Infrastructure and Hardware

Slurm cluster with 8 x H100 Nvidia GPUs.