(self)
| 39 | |
| 40 | class Sampler: |
| 41 | def __init__(self): |
| 42 | self.mesh = VideoLLaMAConfig.get_jax_mesh(FLAGS.mesh_dim) |
| 43 | self.vqgan = VQGAN(FLAGS.vqgan_checkpoint, replicate=False) |
| 44 | self.prefix_tokenizer = AutoTokenizer.from_pretrained(FLAGS.tokenizer, truncation_side='left', padding_side='left') |
| 45 | self.tokenizer = AutoTokenizer.from_pretrained(FLAGS.tokenizer) |
| 46 | self.n_tokens_per_frame = 257 |
| 47 | self.min_buffer_size = 256 |
| 48 | self.sharded_rng = next_rng() |
| 49 | self._load_model() |
| 50 | |
| 51 | @property |
| 52 | def block_size(self): |
nothing calls this directly
no test coverage detected