Update app.py
Browse files
app.py
CHANGED
|
@@ -225,7 +225,7 @@ def infer(
|
|
| 225 |
upload_to_ftp(sd35_path)
|
| 226 |
|
| 227 |
# Convert the generated image to a tensor
|
| 228 |
-
generated_image_tensor = torch.tensor([np.array(sd_image).transpose(2, 0, 1)]).to('cuda') / 255.0
|
| 229 |
# Encode the generated image into latents
|
| 230 |
#with torch.no_grad():
|
| 231 |
# generated_latents = pipe.vae.encode(generated_image_tensor.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
|
|
|
|
| 225 |
upload_to_ftp(sd35_path)
|
| 226 |
|
| 227 |
# Convert the generated image to a tensor
|
| 228 |
+
#generated_image_tensor = torch.tensor([np.array(sd_image).transpose(2, 0, 1)]).to('cuda') / 255.0
|
| 229 |
# Encode the generated image into latents
|
| 230 |
#with torch.no_grad():
|
| 231 |
# generated_latents = pipe.vae.encode(generated_image_tensor.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
|