Iliass Lasri commited on
Commit
3dde997
·
1 Parent(s): 5149d74

changed ckpt

Browse files
Files changed (1) hide show
  1. fspen/src/test.py +1 -1
fspen/src/test.py CHANGED
@@ -4,7 +4,7 @@ from lightning import LightningModule
4
  from models.fspen import FSPEN
5
 
6
  REPO_ID = "iliasslasri/fspen"
7
- FILENAME = "epoch_009.ckpt"
8
  from huggingface_hub import hf_hub_download
9
 
10
 
 
4
  from models.fspen import FSPEN
5
 
6
  REPO_ID = "iliasslasri/fspen"
7
+ FILENAME = "last_ckpt.ckpt"
8
  from huggingface_hub import hf_hub_download
9
 
10