We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 641b53b commit 0daf5eeCopy full SHA for 0daf5ee
predict.py
@@ -2,15 +2,14 @@
2
from typing import List
3
4
import torch
5
+from cog import BasePredictor, Input, Path
6
from diffusers import (
- StableDiffusionPipeline,
7
DDIMScheduler,
8
- EulerDiscreteScheduler,
9
DPMSolverMultistepScheduler,
+ EulerDiscreteScheduler,
10
+ StableDiffusionPipeline,
11
)
12
-from cog import BasePredictor, Input, Path
13
-
14
MODEL_ID = "stabilityai/stable-diffusion-2-1"
15
MODEL_CACHE = "diffusers-cache"
16
@@ -115,4 +114,3 @@ def make_scheduler(name, config):
115
114
"K_EULER": EulerDiscreteScheduler.from_config(config),
116
"DPMSolverMultistep": DPMSolverMultistepScheduler.from_config(config),
117
}[name]
118
0 commit comments