Spaces:
Running
on
Zero
Running
on
Zero
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)
|