Futuretop commited on
Commit
09b7848
·
verified ·
1 Parent(s): c9314ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -35,10 +35,16 @@ import torch
35
  import random
36
  from PIL import Image
37
  from transformers import AutoProcessor, AutoModelForCausalLM
 
 
 
 
 
38
  import diffusers
39
  from diffusers import StableDiffusionPipeline
40
  from diffusers import DiffusionPipeline
41
  from diffusers import EulerDiscreteScheduler
 
42
  diffusers.FlowMatchEulerDiscreteScheduler = EulerDiscreteScheduler
43
 
44
  import transformers.utils.import_utils as _import_utils
 
35
  import random
36
  from PIL import Image
37
  from transformers import AutoProcessor, AutoModelForCausalLM
38
+ from transformers import (
39
+ CLIPTextModel,
40
+ CLIPTokenizer,
41
+ CLIPFeatureExtractor,
42
+ )
43
  import diffusers
44
  from diffusers import StableDiffusionPipeline
45
  from diffusers import DiffusionPipeline
46
  from diffusers import EulerDiscreteScheduler
47
+ from diffusers import UNet2DConditionModel
48
  diffusers.FlowMatchEulerDiscreteScheduler = EulerDiscreteScheduler
49
 
50
  import transformers.utils.import_utils as _import_utils