canonica1 commited on
Commit
c4ba34e
·
verified ·
1 Parent(s): cb09f4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -12
app.py CHANGED
@@ -1,15 +1,3 @@
1
- # -*- coding: utf-8 -*-
2
- """rl_training.ipynb
3
-
4
- Automatically generated by Colab.
5
-
6
- Original file is located at
7
- https://colab.research.google.com/drive/1LJmxNlZNnQCGQOFJYCr-KcA7Q-uvk7gK
8
- """
9
-
10
- !pip install -qqq datasets==3.2.0 transformers==4.47.1 trl==0.14.0 peft==0.14.0 accelerate==1.2.1 bitsandbytes==0.45.2 wandb==0.19.7 --progress-bar off
11
- !pip install -qqq flash-attn --no-build-isolation --progress-bar off
12
-
13
  import torch
14
  from datasets import load_dataset
15
  from peft import LoraConfig, get_peft_model
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
  from datasets import load_dataset
3
  from peft import LoraConfig, get_peft_model