Futuretop commited on
Commit
b6bfe0d
·
verified ·
1 Parent(s): 9d32382

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ import torch
18
  import random
19
  from PIL import Image
20
  from transformers import AutoProcessor, AutoModelForCausalLM
21
- from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 
22
 
23
  REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
24
 
 
18
  import random
19
  from PIL import Image
20
  from transformers import AutoProcessor, AutoModelForCausalLM
21
+ from diffusers import DiffusionPipeline
22
+ from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
23
 
24
  REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
25