Update inference_v42.py
Browse files- inference_v42.py +2 -2
inference_v42.py
CHANGED
|
@@ -28,8 +28,8 @@ zoo.load_clip()
|
|
| 28 |
zoo.load_t5()
|
| 29 |
model = zoo.load_tinyflux(
|
| 30 |
source="AbstractPhil/tiny-flux-deep",
|
| 31 |
-
load_ema=
|
| 32 |
-
ema_path="
|
| 33 |
)
|
| 34 |
print("✓ Models loaded")
|
| 35 |
|
|
|
|
| 28 |
zoo.load_t5()
|
| 29 |
model = zoo.load_tinyflux(
|
| 30 |
source="AbstractPhil/tiny-flux-deep",
|
| 31 |
+
load_ema=True, # this enables loading the checkpoint from the ema path
|
| 32 |
+
ema_path="checkpoints/step_409244.safetensors",
|
| 33 |
)
|
| 34 |
print("✓ Models loaded")
|
| 35 |
|