OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-03 20:22:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix linter issues

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/sd_models_config.py
View File

@ -96,7 +96,6 @@ def guess_model_config_from_state_dict(sd, filename):
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

2
modules/xlmr_m18.py
View File

@ -1,4 +1,4 @@
from transformers import BertPreTrainedModel,BertModel,BertConfig
from transformers import BertPreTrainedModel,BertConfig
import torch.nn as nn
import torch
from transformers.models.xlm_roberta.configuration_xlm_roberta import XLMRobertaConfig