fix linter issues

This commit is contained in:
superhero-7
2023-10-01 12:25:19 +08:00
parent f8f4ff2bb8
commit 2d947175b9
3 changed files with 9 additions and 10 deletions

View File

@@ -95,8 +95,7 @@ def guess_model_config_from_state_dict(sd, filename):
if diffusion_model_input.shape[1] == 8:
return config_instruct_pix2pix
# import pdb; pdb.set_trace()
if sd.get('cond_stage_model.roberta.embeddings.word_embeddings.weight', None) is not None:
if sd.get('cond_stage_model.transformation.weight').size()[0] == 1024:
return config_alt_diffusion_m18