rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Patch safetensors again

Browse Source
This commit is contained in:
somebody
2023-07-07 14:54:40 -05:00
parent 35f3687667
commit 802929f5f2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modeling/lazy_loader.py
View File

@@ -520,8 +520,8 @@ def use_lazy_load(
torch.load = torch_load
#if HAS_SAFETENSORS:
#patch_safetensors(callback)
if HAS_SAFETENSORS:
patch_safetensors(callback)
if dematerialized_modules:
# Most devices can just use Accelerate's implementation, but the Transformers on