spate141
/ grpo_demo.py
Last active
February 2, 2025 02:22
— forked from willccbb/grpo_demo.py
GRPO Llama-1B
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # train_grpo.py - For single 4090 24GB PEFT config | |
| # Import packages | |
| import os | |
| import re | |
| import torch | |
| import random | |
| from datasets import load_dataset | |
| from transformers import AutoTokenizer, AutoModelForCausalLM |
spate141
/ install-tensorflow.sh
Created
October 18, 2016 21:58
— forked from erikbern/install-tensorflow.sh
Installing TensorFlow on EC2
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Note – this is not a bash script (some of the steps require reboot) | |
| # I named it .sh just so Github does correct syntax highlighting. | |
| # | |
| # This is also available as an AMI in us-east-1 (virginia): ami-cf5028a5 | |
| # | |
| # The CUDA part is mostly based on this excellent blog post: | |
| # http://tleyden.github.io/blog/2014/10/25/cuda-6-dot-5-on-aws-gpu-instance-running-ubuntu-14-dot-04/ | |
| # Install various packages | |
| sudo apt-get update |