Update inference_v42.py
Browse files- inference_v42.py +1 -1
inference_v42.py
CHANGED
|
@@ -28,7 +28,7 @@ zoo.load_clip()
|
|
| 28 |
zoo.load_t5()
|
| 29 |
model = zoo.load_tinyflux(
|
| 30 |
source="AbstractPhil/tiny-flux-deep",
|
| 31 |
-
load_ema=
|
| 32 |
ema_path="checkpoints/step_409244.safetensors",
|
| 33 |
)
|
| 34 |
print("✓ Models loaded")
|
|
|
|
| 28 |
zoo.load_t5()
|
| 29 |
model = zoo.load_tinyflux(
|
| 30 |
source="AbstractPhil/tiny-flux-deep",
|
| 31 |
+
load_ema=False, # this enables loading the checkpoint from the ema path, leave false for now to play with the most recent good version
|
| 32 |
ema_path="checkpoints/step_409244.safetensors",
|
| 33 |
)
|
| 34 |
print("✓ Models loaded")
|