sachit-menon commited on
Commit
18f8491
·
verified ·
1 Parent(s): d21eb1b

Update sd_model.py

Browse files
Files changed (1) hide show
  1. sd_model.py +3 -2
sd_model.py CHANGED
@@ -79,8 +79,9 @@ class SDModel(ModelMixin, ConfigMixin, PushToHubMixin):
79
  def __init__(self, cfg: SDModelConfig = None) -> None:
80
  super().__init__()
81
  if cfg is not None: # workaround for default
82
- cfg = SDM
83
- self.cfg = cfg
 
84
  self.noise_scheduler = DDPMScheduler.from_pretrained(
85
  self.cfg.pretrained_model_name_or_path,
86
  subfolder="scheduler",
 
79
  def __init__(self, cfg: SDModelConfig = None) -> None:
80
  super().__init__()
81
  if cfg is not None: # workaround for default
82
+ cfg = SDModelConfig()
83
+ else:
84
+ self.cfg = cfg
85
  self.noise_scheduler = DDPMScheduler.from_pretrained(
86
  self.cfg.pretrained_model_name_or_path,
87
  subfolder="scheduler",