update requirements

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-03-25 10:16:46 -04:00
parent 90f887ac4a
commit c0a2ac9c23
3 changed files with 5 additions and 5 deletions
+1 -1
View File
@@ -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):
+1 -1
View File
@@ -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
View File
@@ -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