suvadityamuk commited on
Commit
c07c20f
·
1 Parent(s): fbb5505

chore: update self-sim flow_model init

Browse files

Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>

Files changed (1) hide show
  1. lib/opt/self_similarity.py +1 -0
lib/opt/self_similarity.py CHANGED
@@ -56,6 +56,7 @@ def optimize_self_similarity(cfg, app, app_type, output_dir):
56
  feats = None
57
 
58
  cond = generation_pipeline.get_cond([app])
 
59
 
60
  if app_type == 'image':
61
  sampler_params = {
 
56
  feats = None
57
 
58
  cond = generation_pipeline.get_cond([app])
59
+ flow_model = generation_pipeline.models['slat_flow_model']
60
 
61
  if app_type == 'image':
62
  sampler_params = {