Futuretop commited on
Commit
a184af3
·
verified ·
1 Parent(s): 3c9d097

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -38,10 +38,8 @@ from transformers import AutoProcessor, AutoModelForCausalLM
38
  import diffusers
39
  from diffusers import StableDiffusionPipeline
40
  from diffusers import DiffusionPipeline
41
- try:
42
- from diffusers import FlowMatchEulerDiscreteScheduler
43
- except ImportError:
44
- from diffusers import EulerDiscreteScheduler as FlowMatchEulerDiscreteScheduler
45
 
46
  import transformers.utils.import_utils as _import_utils
47
  from transformers.utils import is_flash_attn_2_available
 
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
45
  from transformers.utils import is_flash_attn_2_available