Chaerin5 commited on
Commit
8ae8c06
·
1 Parent(s): 85dfb6a

enable zerogpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -262,7 +262,7 @@ def make_ref_cond(
262
  print("ready to run autoencoder")
263
  # print(f"image.device: {image.device}, type(image): {type(image)}")
264
  # image = image.to("cuda")
265
- print(f"autoencoder.encoder.device: {autoencoder.encoder.device}")
266
  latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
267
  return image[None, ...], latent
268
 
 
262
  print("ready to run autoencoder")
263
  # print(f"image.device: {image.device}, type(image): {type(image)}")
264
  # image = image.to("cuda")
265
+ print(f"autoencoder device: {next(autoencoder.parameters()).device}")
266
  latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
267
  return image[None, ...], latent
268