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

fix linter issues

parent f8f4ff2b
......@@ -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
......
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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment