JuyeopDang commited on
Commit
8b67aaf
·
verified ·
1 Parent(s): 8a1772b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -38,6 +38,7 @@ if __name__ == "__main__":
38
  CONFIG_PATH = 'configs/composite_config.yaml'
39
 
40
  vae = VariationalAutoEncoder(CONFIG_PATH)
 
41
  clip = KoCLIPWrapper()
42
  cond_encoder = CLIPEncoder(clip, CONFIG_PATH)
43
  network = UnetWrapper(Unet, CONFIG_PATH, cond_encoder)
 
38
  CONFIG_PATH = 'configs/composite_config.yaml'
39
 
40
  vae = VariationalAutoEncoder(CONFIG_PATH)
41
+ sampler = DDIM(CONFIG_PATH)
42
  clip = KoCLIPWrapper()
43
  cond_encoder = CLIPEncoder(clip, CONFIG_PATH)
44
  network = UnetWrapper(Unet, CONFIG_PATH, cond_encoder)