Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
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, ...]
|
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):
|