sjtu-deepvision commited on
Commit
2fb6499
·
verified ·
1 Parent(s): 7c4abde

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,15 +1,12 @@
1
  import gradio as gr
2
  from PIL import Image
3
- from DAI.pipeline_all import DAIPipeline
 
4
  import os
5
  import tempfile
6
  import numpy as np
7
  import torch as torch
8
  torch.backends.cuda.matmul.allow_tf32 = True
9
- import spaces
10
- import functools
11
-
12
-
13
 
14
  from diffusers import (
15
  AutoencoderKL,
@@ -18,6 +15,8 @@ from diffusers import (
18
 
19
  from transformers import CLIPTextModel, AutoTokenizer
20
 
 
 
21
  from DAI.controlnetvae import ControlNetVAEModel
22
 
23
  from DAI.decoder import CustomAutoencoderKL
 
1
  import gradio as gr
2
  from PIL import Image
3
+ import spaces
4
+ import functools
5
  import os
6
  import tempfile
7
  import numpy as np
8
  import torch as torch
9
  torch.backends.cuda.matmul.allow_tf32 = True
 
 
 
 
10
 
11
  from diffusers import (
12
  AutoencoderKL,
 
15
 
16
  from transformers import CLIPTextModel, AutoTokenizer
17
 
18
+ from DAI.pipeline_all import DAIPipeline
19
+
20
  from DAI.controlnetvae import ControlNetVAEModel
21
 
22
  from DAI.decoder import CustomAutoencoderKL