rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

More details on load failure

Browse Source
This commit is contained in:
somebody
2022-12-09 15:38:27 -06:00
parent 283a9bed7e
commit f4c51b13e8
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
koboldai_settings.py
View File

@@ -929,7 +929,7 @@ class story_settings(settings):
if self.story_id == j["story_id"]: if self.story_id == j["story_id"]:
break break
except FileNotFoundError: except FileNotFoundError:
raise FileNotFoundError("Malformed save file: Missing story.json") raise FileNotFoundError(f"Malformed save file: Missing story.json in {self.save_paths.base}")
disambiguator += 1 disambiguator += 1
self.save_paths.base = os.path.join("stories", save_name + (f" ({disambiguator})" if disambiguator else "")) self.save_paths.base = os.path.join("stories", save_name + (f" ({disambiguator})" if disambiguator else ""))
@@ -946,7 +946,13 @@ class story_settings(settings):
logger.info("Migrating v2 save") logger.info("Migrating v2 save")
with open(v2_path, "r") as file: with open(v2_path, "r") as file:
v2j = json.load(file) v2j = json.load(file)
assert v2j["story_id"] == self.story_id
try:
assert v2j["story_id"] == self.story_id
except AssertionError:
logger.error(f"Story mismatch in v2 migration! Existing file had story id {v2j['story_id']} but we have {self.story_id}")
raise
shutil.move(v2_path, os.path.join(self.save_paths.base, ".v2_old.json")) shutil.move(v2_path, os.path.join(self.save_paths.base, ".v2_old.json"))
with open(self.save_paths.story, "w") as file: with open(self.save_paths.story, "w") as file: