Error on "Semantic_Style_Transfer" Notebook

[krummrey]

Even without changing anything the notebook aborts with a

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
[<ipython-input-12-9f92447fdec0>](https://localhost:8080/#) in <module>()
      1 #Visualize the semantic segment we're going to use
----> 2 agg = aggregate(sample(source, BoxCropper(), perceptor, preprocess, n = mask_samples), labels, perceptor)
      3 visualise(source, agg)
      4 
      5 #Save this as mask.png

[<ipython-input-7-3aed2d19074d>](https://localhost:8080/#) in aggregate(samples, labels, model)
     14             image_features = embedding / embedding.norm(dim=-1, keepdim=True)
     15             logit_scale = model.logit_scale.exp().to(image_features.device)
---> 16             logits_per_image = logit_scale * image_features @ text_features.t()
     17             dist = logits_per_image.float().exp().item()
     18             # dist = spherical_dist(text_emb.float(), embedding.float()).item()

RuntimeError: "dot" not implemented for 'Half'

Any chance to get it to run again?