HReynaud commited on
Commit
166e2c6
·
1 Parent(s): 153ce0d
Files changed (1) hide show
  1. demo.py +1 -0
demo.py CHANGED
@@ -317,6 +317,7 @@ def decode_images(latents):
317
  numpy array of decoded images in [B, H, W, 3] format for single image
318
  or [B, C, T, H, W] for sequences
319
  """
 
320
  if latents is None:
321
  return None
322
 
 
317
  numpy array of decoded images in [B, H, W, 3] format for single image
318
  or [B, C, T, H, W] for sequences
319
  """
320
+ global vae
321
  if latents is None:
322
  return None
323