mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
+1
-1
@@ -1302,7 +1302,7 @@ def check_ui(ver):
|
||||
def same(ver):
|
||||
core = ver['branch'] if ver is not None and 'branch' in ver else 'unknown'
|
||||
ui = ver['ui'] if ver is not None and 'ui' in ver else 'unknown'
|
||||
return core == ui or (core == 'master' and ui == 'main')
|
||||
return (core == ui) or (core == 'master' and ui == 'main') or (core == 'dev' and ui == 'dev')
|
||||
|
||||
t_start = time.time()
|
||||
if not same(ver):
|
||||
|
||||
@@ -127,7 +127,7 @@ class OffloadHook(accelerate.hooks.ModelHook):
|
||||
self.cpu = int(shared.cpu_memory * shared.opts.diffusers_offload_max_cpu_memory * 1024*1024*1024)
|
||||
self.offload_map = {}
|
||||
self.param_map = {}
|
||||
gpu = f'{shared.gpu_memory * shared.opts.diffusers_offload_min_gpu_memory:.3f}-{shared.gpu_memory * shared.opts.diffusers_offload_max_gpu_memory}:{shared.gpu_memory}'
|
||||
gpu = f'{(shared.gpu_memory * shared.opts.diffusers_offload_min_gpu_memory):.2f}-{(shared.gpu_memory * shared.opts.diffusers_offload_max_gpu_memory):.2f}:{shared.gpu_memory:.2f}'
|
||||
shared.log.info(f'Offload: type=balanced op=init watermark={self.min_watermark}-{self.max_watermark} gpu={gpu} cpu={shared.cpu_memory:.3f} limit={shared.opts.cuda_mem_fraction:.2f}')
|
||||
self.validate()
|
||||
super().__init__()
|
||||
|
||||
+3
-3
@@ -32,7 +32,7 @@ pi-heif
|
||||
|
||||
# versioned
|
||||
rich==13.9.4
|
||||
safetensors==0.5.2
|
||||
safetensors==0.5.3
|
||||
tensordict==0.1.2
|
||||
peft==0.14.0
|
||||
httpx==0.24.1
|
||||
@@ -51,8 +51,8 @@ numpy==1.26.4
|
||||
numba==0.59.1
|
||||
protobuf==4.25.3
|
||||
pytorch_lightning==1.9.4
|
||||
tokenizers==0.21.0
|
||||
transformers==4.49.0
|
||||
tokenizers==0.21.1
|
||||
transformers==4.50.0
|
||||
urllib3==1.26.19
|
||||
Pillow==10.4.0
|
||||
timm==0.9.16
|
||||
|
||||
Reference in New Issue
Block a user