ui_extra_networks_hypernets.py 1.0 KB

12345678910111213141516171819202122232425262728293031323334
  1. import os
  2. from modules import shared, ui_extra_networks
  3. class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
  4. def __init__(self):
  5. super().__init__('Hypernetworks')
  6. def refresh(self):
  7. shared.reload_hypernetworks()
  8. def list_items(self):
  9. for name, path in shared.hypernetworks.items():
  10. path, ext = os.path.splitext(path)
  11. previews = [path + ".png", path + ".preview.png"]
  12. preview = None
  13. for file in previews:
  14. if os.path.isfile(file):
  15. preview = "./file=" + file.replace('\\', '/') + "?mtime=" + str(os.path.getmtime(file))
  16. break
  17. yield {
  18. "name": name,
  19. "filename": path,
  20. "preview": preview,
  21. "prompt": f"<hypernet:{name}:1.0>",
  22. "local_preview": path + ".png",
  23. }
  24. def allowed_directories_for_previews(self):
  25. return [shared.cmd_opts.hypernetwork_dir]