Spaces:
Paused
Paused
Commit
·
c07c20f
1
Parent(s):
fbb5505
chore: update self-sim flow_model init
Browse filesSigned-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>
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 = {
|