reference models add size and release date info

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

CHANGELOG.md

@@ -60,6 +60,7 @@
   - **api** new endpoint POST `/sdapi/v1/civitai` to trigger civitai models metadata update  
     accepts optional `page` parameter to search specific networks page  
   - **reference models** additional example images, thanks @liutyi  
+  - **reference models** add model size and release date, thanks @alerikaisattera  
   - **video** support for configurable multi-stage models such as WAN-2.2-14B  
   - **video** new LTX model selection  
   - replace `pynvml` with `nvidia-ml-py` for gpu monitoring  

modules/ui_extra_networks.py

@@ -828,6 +828,11 @@ def create_ui(container, button_parent, tabname, skip_indexing = False):
 
         if is_valid:
             stat_size, stat_mtime = modelstats.stat(item.filename)
+            if hasattr(item, 'size') and item.size > 0:
+                stat_size = item.size
+            if hasattr(item, 'mtime') and item.mtime is not None:
+                stat_mtime = item.mtime
+            print('HERE', item.name, stat_size, stat_mtime)
             desc = item.description
             fullinfo = shared.readfile(os.path.splitext(item.filename)[0] + '.json', silent=True)
             if 'modelVersions' in fullinfo: # sanitize massive objects

modules/ui_extra_networks_checkpoints.py

@@ -2,6 +2,7 @@ import os
 import html
 import json
 import concurrent
+from datetime import datetime
 from modules import shared, ui_extra_networks, sd_models, modelstats, paths
 
 
@@ -45,8 +46,16 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
                     continue
             preview = v.get('preview', v['path'])
             preview_file = self.find_preview_file(os.path.join(paths.reference_path, preview))
-            _size, mtime = modelstats.stat(preview_file)
             name = os.path.normpath(os.path.join(paths.reference_path, k)).replace('\\', '/')
+            size = int(float(v.get('size', 0)) * 1024 * 1024 * 1024)
+            mtime = v.get('date', None)
+            if mtime is None:
+                _size, mtime = modelstats.stat(preview_file)
+            else:
+                try:
+                    mtime = datetime.strptime(mtime, '%Y %B') # 2025 January
+                except Exception:
+                    _size, mtime = modelstats.stat(preview_file)
             yield {
                 "type": 'Model',
                 "name": name,
@@ -57,7 +66,7 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
                 "onclick": '"' + html.escape(f"selectReference({json.dumps(url)})") + '"',
                 "hash": None,
                 "mtime": mtime,
-                "size": 0,
+                "size": size,
                 "info": {},
                 "metadata": {},
                 "description": v.get('desc', ''),