add mtime to reference models

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/models_hf.py

@@ -8,6 +8,7 @@ def hf_init():
     os.environ.setdefault('HF_HUB_DISABLE_SYMLINKS_WARNING', '1')
     os.environ.setdefault('HF_HUB_DISABLE_IMPLICIT_TOKEN', '1')
     os.environ.setdefault('HUGGINGFACE_HUB_VERBOSITY', 'warning')
+    os.environ.setdefault('HF_ENABLE_PARALLEL_LOADING', 'true')
 
 
 def hf_search(keyword):

modules/modelstats.py

@@ -14,7 +14,7 @@ def walk(folder: str):
 
 def stat(fn: str):
     if fn is None or len(fn) == 0 or not os.path.exists(fn):
-        return 0, None
+        return 0, datetime.fromtimestamp(0)
     fs_stat = os.stat(fn, follow_symlinks=False)
     mtime = datetime.fromtimestamp(fs_stat.st_mtime).replace(microsecond=0)
     if os.path.islink(fn):

modules/shared.py

@@ -142,6 +142,7 @@ options_templates.update(options_section(('sd', "Model Loading"), {
 
     "advanced_sep": OptionInfo("<h2>Advanced Options</h2>", "", gr.HTML),
     "sd_checkpoint_autoload": OptionInfo(True, "Model auto-load on start"),
+    "sd_parallel_load": OptionInfo(True, "Model auto-load on start"),
     "sd_checkpoint_autodownload": OptionInfo(True, "Model auto-download on demand"),
     "stream_load": OptionInfo(False, "Model load using streams", gr.Checkbox),
     "diffusers_to_gpu": OptionInfo(False, "Model load model direct to GPU"),

modules/ui_extra_networks.py

@@ -283,6 +283,7 @@ class ExtraNetworksPage:
         htmls = []
 
         if len(self.items) > 0 and self.items[0].get('mtime', None) is not None:
+            shared.opts.extra_networks_sort = 'Date [Newest]'
             if shared.opts.extra_networks_sort == 'Default':
                 pass
             elif shared.opts.extra_networks_sort == 'Name [A-Z]':

modules/ui_extra_networks_checkpoints.py

@@ -26,16 +26,18 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
                 else:
                     continue
             preview = v.get('preview', v['path'])
+            preview_file = self.find_preview_file(os.path.join(reference_dir, preview))
+            _size, mtime = modelstats.stat(preview_file)
             yield {
                 "type": 'Model',
                 "name": os.path.join(reference_dir, k),
                 "title": os.path.join(reference_dir, k),
                 "filename": url,
                 "preview": self.find_preview(os.path.join(reference_dir, preview)),
-                "local_preview": self.find_preview_file(os.path.join(reference_dir, preview)),
+                "local_preview": preview_file,
                 "onclick": '"' + html.escape(f"selectReference({json.dumps(url)})") + '"',
                 "hash": None,
-                "mtime": 0,
+                "mtime": mtime,
                 "size": 0,
                 "info": {},
                 "metadata": {},

modules/ui_extra_networks_styles.py

@@ -1,6 +1,7 @@
 import os
 import html
 import json
+from datetime import datetime
 from modules import shared, extra_networks, ui_extra_networks, styles
 
 
@@ -90,7 +91,7 @@ class ExtraNetworksPageStyles(ui_extra_networks.ExtraNetworksPage):
                 "wildcards": getattr(style, 'wildcards', ''),
                 "local_preview": f"{fn}.{shared.opts.samples_format}",
                 "onclick": '"' + html.escape(f"""return selectStyle({json.dumps(name)})""") + '"',
-                "mtime": getattr(style, 'mtime', 0),
+                "mtime": getattr(style, 'mtime', datetime.fromtimestamp(0)),
                 "size": os.path.getsize(style.filename),
             }
         except Exception as e: