mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
add offload mode to controlnets and framepack
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user