peterpeter8585 commited on
Commit
94715e7
1 Parent(s): 67547b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import random
6
  from diffusers import DiffusionPipeline
7
  import torch
8
  import transformers
9
- from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig
10
  transformers.utils.move_cache()
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  import os
 
6
  from diffusers import DiffusionPipeline
7
  import torch
8
  import transformers
9
+ from transformers import AutoModelForVision2Seq, AutoProcessor
10
  transformers.utils.move_cache()
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  import os