Skip to content

Commit 0daf5ee

Browse files
committed
Sort imports
1 parent 641b53b commit 0daf5ee

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

predict.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,14 @@
22
from typing import List
33

44
import torch
5+
from cog import BasePredictor, Input, Path
56
from diffusers import (
6-
StableDiffusionPipeline,
77
DDIMScheduler,
8-
EulerDiscreteScheduler,
98
DPMSolverMultistepScheduler,
9+
EulerDiscreteScheduler,
10+
StableDiffusionPipeline,
1011
)
1112

12-
from cog import BasePredictor, Input, Path
13-
1413
MODEL_ID = "stabilityai/stable-diffusion-2-1"
1514
MODEL_CACHE = "diffusers-cache"
1615

@@ -115,4 +114,3 @@ def make_scheduler(name, config):
115114
"K_EULER": EulerDiscreteScheduler.from_config(config),
116115
"DPMSolverMultistep": DPMSolverMultistepScheduler.from_config(config),
117116
}[name]
118-

0 commit comments

Comments
 (0)