Spaces:

AbstractPhil
/

tinyflux-lailah

Sleeping

AbstractPhil commited on Jan 25

Commit

ad20a04

verified ·

1 Parent(s): 92345e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -301,7 +301,7 @@ config = TinyFluxDeepConfig()
 model = TinyFluxDeep(config)
 # Load EMA weights (best quality)
-weights_path = hf_hub_download("AbstractPhil/tiny-flux-deep", "checkpoints/step_286250_ema.safetensors")
 weights = load_file(weights_path)
 model.load_state_dict(weights, strict=False)
 model.eval()

 model = TinyFluxDeep(config)
 # Load EMA weights (best quality)
+weights_path = hf_hub_download("AbstractPhil/tiny-flux-deep", "checkpoints/step_297500_ema.safetensors")
 weights = load_file(weights_path)
 model.load_state_dict(weights, strict=False)
 model.eval()