From c0a2ac9c23e68e8a658d7e844d4ec646b6908f27 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Tue, 25 Mar 2025 10:16:46 -0400
Subject: [PATCH] update requirements

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 installer.py          | 2 +-
 modules/sd_offload.py | 2 +-
 requirements.txt      | 6 +++---
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/installer.py b/installer.py
index 006603abe..cb80e3962 100644
--- a/installer.py
+++ b/installer.py
@@ -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):
diff --git a/modules/sd_offload.py b/modules/sd_offload.py
index cadfd3c02..8ab114e93 100644
--- a/modules/sd_offload.py
+++ b/modules/sd_offload.py
@@ -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__()
diff --git a/requirements.txt b/requirements.txt
index e64ada91b..28726d758 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -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