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

Update sd_model.py

Browse files
Files changed (1) hide show
  1. sd_model.py +4 -0
sd_model.py CHANGED
@@ -78,10 +78,14 @@ class SDModelConfig(BaseModelConfig):
78
  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
  self.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",
 
78
  class SDModel(ModelMixin, ConfigMixin, PushToHubMixin):
79
  def __init__(self, cfg: SDModelConfig = None) -> None:
80
  super().__init__()
81
+
82
  if cfg is not None: # workaround for default
83
  self.cfg = SDModelConfig()
84
  else:
85
  self.cfg = cfg
86
+
87
+ print(self.cfg)
88
+
89
  self.noise_scheduler = DDPMScheduler.from_pretrained(
90
  self.cfg.pretrained_model_name_or_path,
91
  subfolder="scheduler",