Pouriarouzrokh commited on
Commit
0f98ba7
·
1 Parent(s): 8a8dac0

tried PLMS5

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def create_ds(img_paths):
49
  ]
50
  return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
51
 
52
- def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="DDIM10"):
53
 
54
  global model
55
  global BASELINE_NOISE
 
49
  ]
50
  return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
51
 
52
+ def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="PLMS5"):
53
 
54
  global model
55
  global BASELINE_NOISE