vilarin commited on
Commit
b20a23b
·
verified ·
1 Parent(s): ab5f65b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -140,6 +140,11 @@ class ModelWrapper:
140
  "text_embeds": batch_pooled_prompt_embeds.squeeze(1)
141
  }
142
 
 
 
 
 
 
143
  eval_images = self.sample(noise=noise, unet_added_conditions=unet_added_conditions, prompt_embed=batch_prompt_embeds, fast_vae_decode=fast_vae_decode)
144
 
145
  end_time = self._get_time()
 
140
  "text_embeds": batch_pooled_prompt_embeds.squeeze(1)
141
  }
142
 
143
+
144
+ print(noise.dtype)
145
+ print(batch_prompt_embeds.dtype)
146
+
147
+
148
  eval_images = self.sample(noise=noise, unet_added_conditions=unet_added_conditions, prompt_embed=batch_prompt_embeds, fast_vae_decode=fast_vae_decode)
149
 
150
  end_time = self._get_time()