Skip to content

autosave on interval / manually, manage session files, and restore session data in oobabooga

Notifications You must be signed in to change notification settings

bekkayya/session_manager

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Session Manager

This was created to fill the void left by the removal of ooba's save/load session, expanded/forked/derived from the Autosave extension with features and options I personally find useful.

Feel free to use/extend/modify. PR fixes welcome.

picture of a user interface with toggles, sliders, and buttons

sessions

save data is broken into session files, which contain checkpoints and a rolling autosave

checkpoints contain the prompt, response, parameters, and model settings

autosaves contain the current state of the input and output text boxes

session files can then have their checkpoint data and parameters loaded manually or automatically

features:

  • optional:
    • save on stop - starting and stopping generation creates a checkpoint (forked feature from AutoSave)
    • auto save - updates the rolling autosave with new data every x seconds
    • auto load - auto loads the last selected, or newest, session when the ooba first loads
    • auto session: creates a new session if the prompt is changed between checkpoints
  • file management - create, select, rename, and delete, sessions and checkpoints from the ui
  • json preview - expandable preview to see the selected checkpoint's raw json
  • custom save location config - output_path in settings.json can be used to overwrite the default save location
  • futureproofish - settings are collected from the internal ooba functions, so new settings and sliders should be automatically saved.

feature todo:

  • save string customization
  • chat & notebook functionality
  • capture cli output, parameters preset

known issues:

  • model parameters are saved but cant be loaded due to I cant figure it out, feature is currently disabled
  • sometimes gradio dropdowns have a different entry ticked than selected
    • manual solution: select something else and return to force an update
  • currently only designed for the default page, notebook might work somewhat, chat probably not at all
  • ui refresh code is clunky in parts mostly because I'm unfamiliar with gradio
  • autosaving and manual checkpointing can lag behind by several tokens
  • checkpointing at the very start before any generation often captures nothing, not much to be done about that

About

autosave on interval / manually, manage session files, and restore session data in oobabooga

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%