Chaerin5 commited on
Commit
6a078f3
·
1 Parent(s): bde36a9

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
  ):
263
  print("ready to run autoencoder")
264
  # print(f"image.device: {image.device}, type(image): {type(image)}")
265
- latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...].to("cuda")).sample()
266
  return image[None, ...], latent
267
 
268
  def get_ref_anno(ref):
 
262
  ):
263
  print("ready to run autoencoder")
264
  # print(f"image.device: {image.device}, type(image): {type(image)}")
265
+ latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
266
  return image[None, ...], latent
267
 
268
  def get_ref_anno(ref):