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

Update inference_v42.py

Browse files
Files changed (1) hide show
  1. 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=False, # leave false for now, this loads incorrectly when true
32
- ema_path="step_409244_ema.safetensors", # ema handles more generic prompts
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