(self, input_images, n_new_frames, n_candidates, temperature=1.0, top_p=1.0)
| 159 | return new_images |
| 160 | |
| 161 | def __call__(self, input_images, n_new_frames, n_candidates, temperature=1.0, top_p=1.0): |
| 162 | output = [] |
| 163 | for seq in input_images: |
| 164 | output.append( |
| 165 | [self.generate_once(seq, n_new_frames, temperature, top_p) |
| 166 | for _ in range(n_candidates)] |
| 167 | ) |
| 168 | return output |
| 169 | |
| 170 | |
| 171 | class MultiProcessInferenceModel(InferenceModel): |
nothing calls this directly
no test coverage detected