WiNE-iNEFF commited on
Commit
cc69909
·
1 Parent(s): 2e62db4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ class Model:
26
  self.code = code
27
 
28
  model = [
29
- Model("DDIMScheduler", "scheduler = DDIMScheduler.from_pretrained(pipeline_name)"),
30
- Model("PNDMScheduler", "scheduler = PNDMScheduler.from_pretrained(pipeline_name)")]
31
 
32
  current_model = model[0]
33
 
 
26
  self.code = code
27
 
28
  model = [
29
+ Model("DDIMScheduler", "DDIMScheduler.from_pretrained(pipeline_name)"),
30
+ Model("PNDMScheduler", "PNDMScheduler.from_pretrained(pipeline_name)")]
31
 
32
  current_model = model[0]
33