prevent accidental creation of CLIP models in float32 type when user wants float16

This commit is contained in:
AUTOMATIC1111
2024-06-16 11:04:19 +03:00
parent 7ee2114cd9
commit b443fdcf76
2 changed files with 4 additions and 3 deletions

View File

@@ -406,6 +406,7 @@ def set_model_fields(model):
if not hasattr(model, 'latent_channels'):
model.latent_channels = 4
def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer):
sd_model_hash = checkpoint_info.calculate_shorthash()
timer.record("calculate hash")