Балаганский Никита Николаевич commited on
Commit
d0289f9
1 Parent(s): 1a2cd56
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -54,7 +54,7 @@ class Generator:
54
  input_prompt,
55
  num_samples,
56
  )
57
- text = st.text()
58
  for i in range(max_length):
59
  is_caif_step = (
60
  i % caif_period == 0 and self.caif_sampler is not None
 
54
  input_prompt,
55
  num_samples,
56
  )
57
+ text = st.empty()
58
  for i in range(max_length):
59
  is_caif_step = (
60
  i % caif_period == 0 and self.caif_sampler is not None