ford442 commited on
Commit
378493f
·
verified ·
1 Parent(s): 0b1f594

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)