add offload mode to controlnets and framepack

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-22 15:40:52 -04:00
parent 3b89094ac1
commit 87fc6c7913
3 changed files with 29 additions and 6 deletions
+6
View File
@@ -205,6 +205,12 @@ class ControlNet():
self.load_config = { 'cache_dir': cache_dir }
if load_config is not None:
self.load_config.update(load_config)
if opts.offline_mode:
self.load_config["local_files_only"] = True
os.environ['HF_HUB_OFFLINE'] = '1'
else:
os.environ.pop('HF_HUB_OFFLINE', None)
os.unsetenv('HF_HUB_OFFLINE')
if model_id is not None:
self.load()