update model funcs

This commit is contained in:
logtd
2024-05-04 16:50:32 -06:00
parent e4d8d14c35
commit 471d86cfc6
+16 -1
View File
@@ -45,9 +45,15 @@ class InstanceConditioning:
def is_clone(self, other, *args, **kwargs):
return other == self
def clone(self):
return self
def model_size(self, *args, **kwargs):
return 0
def memory_required(self, *args, **kwargs):
return 0
def model_patches_to(self, device_or_dtype, *args, **kwargs):
if device_or_dtype == torch.float16 or device_or_dtype == torch.float32:
return
@@ -60,8 +66,17 @@ class InstanceConditioning:
def model_dtype(self, *args, **kwargs):
return torch.float32
def patch_model(self, device_to, *args, **kwargs):
def patch_model(self, *args, **kwargs):
return
def unpatch_weights(self, *args, **kwargs):
return
def unpatch_model(self, *args, **kwargs):
return
def set_model_patch(self, *args, **kwargs):
return
def set_model_patch_replace(self, *args, **kwargs):
return