Spaces:
Running
Running
kovacsvi
commited on
Commit
·
73ee479
1
Parent(s):
75bb7a1
debug print for minor probs
Browse files
interfaces/cap_minor_media.py
CHANGED
@@ -118,7 +118,9 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
|
|
118 |
with torch.no_grad():
|
119 |
minor_logits = minor_model(**inputs).logits
|
120 |
minor_probs = F.softmax(minor_logits, dim=-1)
|
121 |
-
|
|
|
|
|
122 |
# Restrict to valid minor codes
|
123 |
valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
|
124 |
filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
|
|
|
118 |
with torch.no_grad():
|
119 |
minor_logits = minor_model(**inputs).logits
|
120 |
minor_probs = F.softmax(minor_logits, dim=-1)
|
121 |
+
|
122 |
+
|
123 |
+
print(minor_probs) # debug
|
124 |
# Restrict to valid minor codes
|
125 |
valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
|
126 |
filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
|