alppo commited on
Commit
fad6fd3
·
1 Parent(s): 421323e

delete some files

Browse files
Files changed (2) hide show
  1. evaluate_module.py +13 -1
  2. vae_model.pth +0 -3
evaluate_module.py CHANGED
@@ -1 +1,13 @@
1
- # to do : evaluate function
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ class Generator:
2
+ def __init__(config, unet, scheduler, embedding):
3
+ self.config = config
4
+ self.unet = unet
5
+ self.scheduler = scheduler
6
+ self.embedding = embedding
7
+
8
+ def generate():
9
+ with torch.no_grad():
10
+ uncond_image = torch.zeros((1, 1, config.image_size, config.image_size), device=device)
11
+ mu, log_var = vae.encode(uncond_image)
12
+ uncond_latent = torch.cat((mu, log_var), dim=1)
13
+ uncond_latent = uncond_latent.unsqueeze(0)
vae_model.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a8e271f76eea5c196e3d2a3b69b3a605610ae53fab76f1b2d451d96d13dfcfcd
3
- size 277897466