enzokro commited on
Commit
ffaff65
·
1 Parent(s): 4945799

bug fixes to api call

Browse files
Files changed (2) hide show
  1. app.ipynb +1 -1
  2. app.py +1 -1
app.ipynb CHANGED
@@ -48,7 +48,7 @@
48
  "model_kwargs = {\n",
49
  " 'better_vae': better_vae,\n",
50
  " 'unet_attn_slice': unet_attn_slice,\n",
51
- " 'sampler_kls': hf_sampler,\n",
52
  "}\n",
53
  "\n",
54
  "def load_model():\n",
 
48
  "model_kwargs = {\n",
49
  " 'better_vae': better_vae,\n",
50
  " 'unet_attn_slice': unet_attn_slice,\n",
51
+ " 'scheduler_kls': hf_sampler,\n",
52
  "}\n",
53
  "\n",
54
  "def load_model():\n",
app.py CHANGED
@@ -32,7 +32,7 @@ hf_sampler = 'dpm_multi'
32
  model_kwargs = {
33
  'better_vae': better_vae,
34
  'unet_attn_slice': unet_attn_slice,
35
- 'sampler_kls': hf_sampler,
36
  }
37
 
38
  def load_model():
 
32
  model_kwargs = {
33
  'better_vae': better_vae,
34
  'unet_attn_slice': unet_attn_slice,
35
+ 'scheduler_kls': hf_sampler,
36
  }
37
 
38
  def load_model():