Upload model
Browse files- modeling_gzipembed.py +1 -1
modeling_gzipembed.py
CHANGED
|
@@ -21,7 +21,7 @@ class GZIPEmbeddingModel(PreTrainedModel):
|
|
| 21 |
global p
|
| 22 |
def calculate_ncd_row(data_row):
|
| 23 |
i = data_row[0]
|
| 24 |
-
row =
|
| 25 |
return i, row
|
| 26 |
if type(prompt) == str:
|
| 27 |
prompt = [prompt]
|
|
|
|
| 21 |
global p
|
| 22 |
def calculate_ncd_row(data_row):
|
| 23 |
i = data_row[0]
|
| 24 |
+
row = self.ncd(data_row[1], prompt)
|
| 25 |
return i, row
|
| 26 |
if type(prompt) == str:
|
| 27 |
prompt = [prompt]
|