RamAnanth1 commited on
Commit
2b80ac9
1 Parent(s): 7093637

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -206,7 +206,7 @@ class Model:
206
  eta=0.0,
207
  x_T=None,
208
  features_adapter1=features_adapter,
209
- mode = 'pose',
210
  con_strength = con_strength)
211
 
212
  x_samples_ddim = self.model.decode_first_stage(samples_ddim)
 
206
  eta=0.0,
207
  x_T=None,
208
  features_adapter1=features_adapter,
209
+ mode = 'sketch',
210
  con_strength = con_strength)
211
 
212
  x_samples_ddim = self.model.decode_first_stage(samples_ddim)