Merge pull request #78 from Haidra-Org/regen

fix: better signal to comfyui lowvram mode is desired
Haidra-Org · Oct 4, 2023 · 15e326e · 15e326e
2 parents 2f8369b + 68281bb
commit 15e326e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hordelib/comfy_horde.py b/hordelib/comfy_horde.py
@@ -146,6 +146,7 @@ def always_cpu(parameters, dtype):
 
         comfy.model_management.unet_inital_load_device = always_cpu
         comfy.model_management.DISABLE_SMART_MEMORY = True
+        comfy.model_management.lowvram_available = True
 
         total_vram = get_torch_total_vram_mb()
         total_ram = psutil.virtual_memory().total / (1024 * 1024)

diff --git a/requirements.txt b/requirements.txt
@@ -12,7 +12,7 @@ torchsde
 einops
 open-clip-torch
 transformers>=4.25.1
-safetensors
+safetensors>=0.3.0
 pytorch_lightning
 pynvml
 aiohttp