Балаганский Никита Николаевич
commited on
Commit
•
6ba5cb1
1
Parent(s):
23b915b
layout fix
Browse files- generator.py +2 -1
generator.py
CHANGED
@@ -71,6 +71,7 @@ class Generator:
|
|
71 |
"hovermode": "x",
|
72 |
|
73 |
})
|
|
|
74 |
for i in range(max_length):
|
75 |
is_caif_step = (
|
76 |
i % caif_period == 0 and self.caif_sampler is not None
|
@@ -97,7 +98,7 @@ class Generator:
|
|
97 |
scatter_data = go.Scatter({
|
98 |
"x": list(range(len(gen_history))),
|
99 |
"y": gen_history,
|
100 |
-
"hovertext": [self.tokenizer.decode(t) for t in input_ids[0]]
|
101 |
})
|
102 |
fig = go.Figure([scatter_data], layout=layout)
|
103 |
plot.plotly_chart(fig, use_container_width=True)
|
|
|
71 |
"hovermode": "x",
|
72 |
|
73 |
})
|
74 |
+
inp_len = len(input_ids[0])
|
75 |
for i in range(max_length):
|
76 |
is_caif_step = (
|
77 |
i % caif_period == 0 and self.caif_sampler is not None
|
|
|
98 |
scatter_data = go.Scatter({
|
99 |
"x": list(range(len(gen_history))),
|
100 |
"y": gen_history,
|
101 |
+
"hovertext": [self.tokenizer.decode(t) for t in input_ids[0][inp_len:]]
|
102 |
})
|
103 |
fig = go.Figure([scatter_data], layout=layout)
|
104 |
plot.plotly_chart(fig, use_container_width=True)
|