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

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,
32
- ema_path="step_407682.safetensors",
33
  )
34
  print("✓ Models loaded")
35
 
 
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