HelloSun commited on
Commit
02fd760
·
verified ·
1 Parent(s): c80353d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -19
app.py CHANGED
@@ -13,6 +13,26 @@ from diffusers import (
13
  EulerDiscreteScheduler,
14
  )
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
 
17
  model_id = "helenai/Linaqruf-anything-v3.0-ov"
18
  num_inference_steps = 25
@@ -52,25 +72,6 @@ def infer(prompt,negative_prompt):
52
  return image
53
 
54
 
55
- def get_scheduler(scheduler_config: Dict, name: str) -> Optional[Callable]:
56
- scheduler_factory_map = {
57
- "DPM++ 2M Karras": lambda: DPMSolverMultistepScheduler.from_config(
58
- scheduler_config, use_karras_sigmas=True
59
- ),
60
- "DPM++ SDE Karras": lambda: DPMSolverSinglestepScheduler.from_config(
61
- scheduler_config, use_karras_sigmas=True
62
- ),
63
- "DPM++ 2M SDE Karras": lambda: DPMSolverMultistepScheduler.from_config(
64
- scheduler_config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"
65
- ),
66
- "Euler": lambda: EulerDiscreteScheduler.from_config(scheduler_config),
67
- "Euler a": lambda: EulerAncestralDiscreteScheduler.from_config(
68
- scheduler_config
69
- ),
70
- "DDIM": lambda: DDIMScheduler.from_config(scheduler_config),
71
- }
72
- return scheduler_factory_map.get(name, lambda: None)()
73
-
74
 
75
 
76
  examples = [
 
13
  EulerDiscreteScheduler,
14
  )
15
 
16
+ def get_scheduler(scheduler_config: Dict, name: str) -> Optional[Callable]:
17
+ scheduler_factory_map = {
18
+ "DPM++ 2M Karras": lambda: DPMSolverMultistepScheduler.from_config(
19
+ scheduler_config, use_karras_sigmas=True
20
+ ),
21
+ "DPM++ SDE Karras": lambda: DPMSolverSinglestepScheduler.from_config(
22
+ scheduler_config, use_karras_sigmas=True
23
+ ),
24
+ "DPM++ 2M SDE Karras": lambda: DPMSolverMultistepScheduler.from_config(
25
+ scheduler_config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++"
26
+ ),
27
+ "Euler": lambda: EulerDiscreteScheduler.from_config(scheduler_config),
28
+ "Euler a": lambda: EulerAncestralDiscreteScheduler.from_config(
29
+ scheduler_config
30
+ ),
31
+ "DDIM": lambda: DDIMScheduler.from_config(scheduler_config),
32
+ }
33
+ return scheduler_factory_map.get(name, lambda: None)()
34
+
35
+
36
 
37
  model_id = "helenai/Linaqruf-anything-v3.0-ov"
38
  num_inference_steps = 25
 
72
  return image
73
 
74
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
75
 
76
 
77
  examples = [