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

Update sd_model.py

Browse files
Files changed (1) hide show
  1. sd_model.py +1 -1
sd_model.py CHANGED
@@ -79,7 +79,7 @@ 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 = SDModelConfig()
83
  else:
84
  self.cfg = cfg
85
  self.noise_scheduler = DDPMScheduler.from_pretrained(
 
79
  def __init__(self, cfg: SDModelConfig = None) -> None:
80
  super().__init__()
81
  if cfg is not None: # workaround for default
82
+ self.cfg = SDModelConfig()
83
  else:
84
  self.cfg = cfg
85
  self.noise_scheduler = DDPMScheduler.from_pretrained(