lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-27 09:41:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Second Attempt for #1502

Browse Source
This commit is contained in:
layerdiffusion
2024-08-28 08:07:42 -07:00
parent 37fcb7bce8
commit 0abb6c4686
4 changed files with 102 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/operations_bnb.py
View File

@@ -92,10 +92,8 @@ class ForgeLoader4Bit(torch.nn.Module):
self.quant_type = quant_type
def _apply(self, fn, recurse=True):
if self.weight is not None:
self.weight = utils.tensor2parameter(fn(self.weight))
if self.bias is not None:
self.bias = utils.tensor2parameter(fn(self.bias))
for k, p in self.named_parameters(recurse=False, remove_duplicate=True):
setattr(self, k, utils.tensor2parameter(fn(p)))
return self
def _save_to_state_dict(self, destination, prefix, keep_vars):