AYYasaswini commited on
Commit
fd9c32c
·
verified ·
1 Parent(s): 81778ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -67,12 +67,7 @@ latents = latents * scheduler.init_noise_sigma # Scaling (previous versions did
67
  with autocast("cuda"): # will fallback to CPU if no CUDA; no autocast for MPS
68
  for i, t in tqdm(enumerate(scheduler.timesteps), total=len(scheduler.timesteps)):
69
 
70
- """It's working, but that's quite a bit of code! Let's look at the components one by one.
71
 
72
- ## The Autoencoder (AE)
73
-
74
- The AE can 'encode' an image into some sort of latent representation, and decode this back into an image. I've wrapped the code for this into a couple of functions here so we can see what this looks like in action:
75
- """
76
 
77
  def pil_to_latent(input_im):
78
  # Single image -> single latent in a batch (so size 1, 4, 64, 64)
 
67
  with autocast("cuda"): # will fallback to CPU if no CUDA; no autocast for MPS
68
  for i, t in tqdm(enumerate(scheduler.timesteps), total=len(scheduler.timesteps)):
69
 
 
70
 
 
 
 
 
71
 
72
  def pil_to_latent(input_im):
73
  # Single image -> single latent in a batch (so size 1, 4, 64, 64)