Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
# Implementation of
|
2 |
|
3 |
import inspect
|
4 |
from typing import Any, Callable, Dict, List, Optional, Tuple, Union
|
@@ -280,7 +280,7 @@ def retrieve_timesteps(
|
|
280 |
timesteps = scheduler.timesteps
|
281 |
return timesteps, num_inference_steps
|
282 |
|
283 |
-
class
|
284 |
DiffusionPipeline,
|
285 |
StableDiffusionMixin,
|
286 |
FromSingleFileMixin,
|
|
|
1 |
+
# Implementation of StableDiffusionXLPipeline
|
2 |
|
3 |
import inspect
|
4 |
from typing import Any, Callable, Dict, List, Optional, Tuple, Union
|
|
|
280 |
timesteps = scheduler.timesteps
|
281 |
return timesteps, num_inference_steps
|
282 |
|
283 |
+
class StableDiffusionXLPipeline(
|
284 |
DiffusionPipeline,
|
285 |
StableDiffusionMixin,
|
286 |
FromSingleFileMixin,
|