Fix cluster to clustering

app.py

@@ -56,7 +56,7 @@ def get_noising_schedule(i, max_it, sharpness=5.0):
     x = i / max_it
     return (np.exp(-sharpness * x) - np.exp(-sharpness)) / (1 - np.exp(-sharpness))
 
-def noisify_answer(input_ids, answer_start, threshold=1.0, eot_weight=1.0, cluster=0.5):
+def noisify_answer(input_ids, answer_start, threshold=1.0, eot_weight=1.0, clustering=0.5):
     noised = input_ids.copy()
     answer_len = len(noised) - answer_start
     num_to_noise = int(threshold * answer_len)
@@ -69,7 +69,7 @@ def noisify_answer(input_ids, answer_start, threshold=1.0, eot_weight=1.0, clust
     mixed_probs /= mixed_probs.sum()
 
     # Determine number of clusters and average cluster size
-    num_clusters = max(1, int((1 - cluster) * num_to_noise))  # fewer clusters if more intensity
+    num_clusters = max(1, int((1 - clustering) * num_to_noise))  # fewer clusters if more intensity
     cluster_size = max(1, int(num_to_noise / num_clusters))
 
     noised_indices = set()