Commit 2d947175 authored by superhero-7's avatar superhero-7
Browse files

fix linter issues

parent f8f4ff2b

modules/sd_models_config.py

+1 −2
Original line number Diff line number Diff line
@@ -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

modules/xlmr_m18.py

+6 −6
Original line number Diff line number Diff line 
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

modules/sd_hijack.py

+2 −2

File changed.

Contains only whitespace changes.

CRA Git | Maintained and supported by SUSTech CRA and CCSE