DR-Rakshitha commited on
Commit
1465e91
·
1 Parent(s): 06e5052

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@
7
  #----------------------------------------------------------------------------------------------------------------------------
8
  # !pip install -q accelerate==0.21.0 peft==0.4.0 bitsandbytes==0.40.2 transformers==4.31.0 trl==0.4.7
9
  import torch
10
- from datasets import load_dataset
11
  from transformers import (
12
  AutoModelForCausalLM,
13
  AutoTokenizer,
 
7
  #----------------------------------------------------------------------------------------------------------------------------
8
  # !pip install -q accelerate==0.21.0 peft==0.4.0 bitsandbytes==0.40.2 transformers==4.31.0 trl==0.4.7
9
  import torch
10
+ # from datasets import load_dataset
11
  from transformers import (
12
  AutoModelForCausalLM,
13
  AutoTokenizer,