Skip to content

Commit f9ba466

Browse files
committed
Store POET weights in user cache directory
Update `poet_default_weights_dir()` to use `~/.cache/dlclivegui/poet` instead of a working-directory temp path. This makes the weights location stable across runs and avoids writing model files into the project directory.
1 parent 6b6a00c commit f9ba466

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

  • dlclivegui/services/inference/models/poet

‎dlclivegui/services/inference/models/poet/weights.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,7 @@
1515

1616

1717
def poet_default_weights_dir() -> Path:
18-
# TODO: move to an app cache rather than a temporary directory
19-
# return Path(tempfile.gettempdir()) / "dlclivegui" / "poet"
20-
return Path.cwd() / "TEMP_WEIGHTS/poet"
18+
return Path.home() / ".cache/dlclivegui/poet"
2119

2220

2321
class WeightsDownloadWorker(QObject):

0 commit comments

Comments
 (0)