Commit 81c650c
committed
Fix text cache source_path_in_name: prompt_path -> image_path
'prompt_path' doesn't exist as a pipeline name, so _get_source_path
returned None for every text cache lookup, causing 100% cache misses
and device mismatch errors when the text encoder was on CPU during
training. Use 'image_path' as the key, matching the old DiskCache
behavior.1 parent 948935d commit 81c650c
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
400 | 400 | | |
401 | 401 | | |
402 | 402 | | |
403 | | - | |
| 403 | + | |
404 | 404 | | |
405 | 405 | | |
406 | 406 | | |
| |||
0 commit comments