Fix generator determinism: forward generator through scheduler steps and seeded noise
Browse files- FiTv1-XL-2-256/pipeline.py +2 -0
- FiTv2-3B-2-512/pipeline.py +1 -1
FiTv1-XL-2-256/pipeline.py
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
Load with native Hugging Face diffusers and trust_remote_code=True.
|
| 3 |
"""
|
| 4 |
|
|
|
|
|
|
|
| 5 |
import importlib
|
| 6 |
import inspect
|
| 7 |
import json
|
|
|
|
| 2 |
Load with native Hugging Face diffusers and trust_remote_code=True.
|
| 3 |
"""
|
| 4 |
|
| 5 |
+
from __future__ import annotations
|
| 6 |
+
|
| 7 |
import importlib
|
| 8 |
import inspect
|
| 9 |
import json
|
FiTv2-3B-2-512/pipeline.py
CHANGED
|
@@ -21,7 +21,7 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineO
|
|
| 21 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
| 22 |
from diffusers.utils.torch_utils import randn_tensor
|
| 23 |
|
| 24 |
-
DEFAULT_NATIVE_RESOLUTION =
|
| 25 |
|
| 26 |
EXAMPLE_DOC_STRING = """
|
| 27 |
Examples:
|
|
|
|
| 21 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
| 22 |
from diffusers.utils.torch_utils import randn_tensor
|
| 23 |
|
| 24 |
+
DEFAULT_NATIVE_RESOLUTION = 256
|
| 25 |
|
| 26 |
EXAMPLE_DOC_STRING = """
|
| 27 |
Examples:
|