devforfu
commited on
Commit
•
5c9258d
1
Parent(s):
b1d3d29
Cast to Path
Browse files- realfake/utils.py +1 -0
realfake/utils.py
CHANGED
|
@@ -114,6 +114,7 @@ def get_image(url: str, filename: Path | None = None):
|
|
| 114 |
|
| 115 |
|
| 116 |
def find_latest_checkpoint(dirname: Path) -> Path:
|
|
|
|
| 117 |
checkpoints = list(dirname.glob("*.ckpt"))
|
| 118 |
if not checkpoints:
|
| 119 |
return None
|
|
|
|
| 114 |
|
| 115 |
|
| 116 |
def find_latest_checkpoint(dirname: Path) -> Path:
|
| 117 |
+
dirname = Path(dirname)
|
| 118 |
checkpoints = list(dirname.glob("*.ckpt"))
|
| 119 |
if not checkpoints:
|
| 120 |
return None
|