Fixed typing bugs in styleclip projection

styleclip/styleclip_global.py

@@ -120,6 +120,7 @@ def get_direction(neutral_class, target_class, beta, di, clip_model=None):
 
     dt = class_weights[:, 1] - class_weights[:, 0]
     dt = dt / dt.norm()
+    dt = dt.type(type(di))
     relevance = di @ dt
     mask = relevance.abs() > beta
     direction = relevance * mask
@@ -151,7 +152,7 @@ def style_dict_to_style_tensor(style_dict, reference_generator):
     return style_tensor
 
 def project_code_with_styleclip(source_latent, source_class, target_class, alpha, beta, reference_generator, di, clip_model=None):
-        edit_direction = get_direction(source_class, target_class, beta)
+        edit_direction = get_direction(source_class, target_class, beta, di, clip_model)
 
         source_s = style_dict_to_style_tensor(source_latent, reference_generator)