AbstractPhil commited on
Commit
e1795a0
·
verified ·
1 Parent(s): 76c9ae3

Update inference_v42.py

Browse files
Files changed (1) hide show
  1. 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=True, # this enables loading the checkpoint from the ema path
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")