dpaul93 commited on
Commit
ffbc1bf
·
verified ·
1 Parent(s): 562ef9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import bitsandbytes as bnb
7
  import torch
8
  import torch.nn as nn
9
  import transformers
 
10
  from datasets import load_dataset, Dataset
11
  from huggingface_hub import notebook_login
12
 
 
7
  import torch
8
  import torch.nn as nn
9
  import transformers
10
+ imprt accelerate
11
  from datasets import load_dataset, Dataset
12
  from huggingface_hub import notebook_login
13