

# DISCLAIMER: This code is strongly influenced by https://github.com/pesser/pytorch_diffusion
# and https://github.com/hojonathanho/diffusion
import math
from dataclasses import dataclass
from typing import List, Optional, Tuple, Union

import numpy as np
import torch

from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.schedulers.scheduling_utils import SchedulerMixin
from diffusers.utils import BaseOutput, deprecate

@dataclass
# Copied from diffusers.schedulers.scheduling_ddpm.DDPMSchedulerOutput with DDPM->DDIM
class     a__  (    snake_case	):
			"""simple docstring"""







			__lowerCamelCase       				=				42
			__lowerCamelCase       				=				None



def     lowerCAmelCase__							(		SCREAMING_SNAKE_CASE_:							Any		,				SCREAMING_SNAKE_CASE_:							Optional[Any]=0.999		,				SCREAMING_SNAKE_CASE_:							str="cosine"		,				)	->					Optional[int]:




				'''simple docstring'''





				if alpha_transform_type == "cosine":

								def alpha_bar_fn(SCREAMING_SNAKE_CASE_:							Optional[int]					):
												return math.cos((t + 0.008) / 1.008 * math.pi / 2					) ** 2

				elif alpha_transform_type == "exp":

								def alpha_bar_fn(SCREAMING_SNAKE_CASE_:							List[Any]					):
												return math.exp(t * -12.0					)

				else:
								raise ValueError(F'Unsupported alpha_tranform_type: {alpha_transform_type}'					)

				A__						=	[]
				for i in range(SCREAMING_SNAKE_CASE_					):
								A__						=	i / num_diffusion_timesteps
								A__						=	(i + 1) / num_diffusion_timesteps
								betas.append(min(1 - alpha_bar_fn(SCREAMING_SNAKE_CASE_					) / alpha_bar_fn(SCREAMING_SNAKE_CASE_					)		,				SCREAMING_SNAKE_CASE_					)					)
				return torch.tensor(SCREAMING_SNAKE_CASE_		,				dtype=torch.floataa					)





class     a__  (    snake_case							,		snake_case	):
			"""simple docstring"""







			__lowerCamelCase       				=				1


			@register_to_config
			def __init__(			self					,					lowercase = 1000					,					lowercase = 0.0001					,					lowercase = 0.02					,					lowercase = "linear"					,					lowercase = None					,					lowercase = True					,					lowercase = True					,					lowercase = 0					,					lowercase = "epsilon"					,					lowercase = 1.0					,					**lowercase					,					)     ->  Union[str, Any]:
							'''simple docstring'''






							if kwargs.get("set_alpha_to_one"					,					lowercase				) is not None:
											A__						=	(
											    "The `set_alpha_to_one` argument is deprecated. Please use `set_alpha_to_zero` instead."
											)
											deprecate("set_alpha_to_one"					,					"1.0.0"					,					lowercase					,					standard_warn=lowercase				)
											A__						=	kwargs["set_alpha_to_one"]
							if trained_betas is not None:
											A__						=	torch.tensor(lowercase					,					dtype=torch.floataa				)
							elif beta_schedule == "linear":
											A__						=	torch.linspace(lowercase					,					lowercase					,					lowercase					,					dtype=torch.floataa				)
							elif beta_schedule == "scaled_linear":
											# this schedule is very specific to the latent diffusion model.
											A__						=	(
											    torch.linspace(beta_start**0.5					,					beta_end**0.5					,					lowercase					,					dtype=torch.floataa				) ** 2
											)
							elif beta_schedule == "squaredcos_cap_v2":
											# Glide cosine schedule
											A__						=	betas_for_alpha_bar(lowercase				)
							else:
											raise NotImplementedError(F'{beta_schedule} does is not implemented for {self.__class__}'				)

							A__						=	1.0 - self.betas
							A__						=	torch.cumprod(self.alphas					,					dim=0				)

							# At every step in inverted ddim, we are looking into the next alphas_cumprod
							# For the final step, there is no next alphas_cumprod, and the index is out of bounds
							# `set_alpha_to_zero` decides whether we set this parameter simply to zero
							# in this case, self.step() just output the predicted noise
							# or whether we use the final alpha of the "non-previous" one.
							A__						=	torch.tensor(0.0				) if set_alpha_to_zero else self.alphas_cumprod[-1]

							# standard deviation of the initial noise distribution
							A__						=	1.0

							# setable values
							A__						=	None
							A__						=	torch.from_numpy(np.arange(0					,					lowercase				).copy().astype(np.intaa				)				)


			def    UpperCamelCase						(			self					,					lowercase					,					lowercase = None				)     ->  torch.FloatTensor:
							'''simple docstring'''






							return sample


			def    UpperCamelCase						(			self					,					lowercase					,					lowercase = None				)     ->  Optional[int]:
							'''simple docstring'''






							if num_inference_steps > self.config.num_train_timesteps:
											raise ValueError(
											    F'`num_inference_steps`: {num_inference_steps} cannot be larger than `self.config.train_timesteps`:'
											    F' {self.config.num_train_timesteps} as the unet model trained with this scheduler can only handle'
											    F' maximal {self.config.num_train_timesteps} timesteps.'				)

							A__						=	num_inference_steps
							A__						=	self.config.num_train_timesteps // self.num_inference_steps
							# creates integer timesteps by multiplying by ratio
							# casting to int to avoid issues when num_inference_step is power of 3
							A__						=	(np.arange(0					,					lowercase				) * step_ratio).round().copy().astype(np.intaa				)
							A__						=	torch.from_numpy(lowercase				).to(lowercase				)
							self.timesteps += self.config.steps_offset


			def    UpperCamelCase						(			self					,					lowercase					,					lowercase					,					lowercase					,					lowercase = 0.0					,					lowercase = False					,					lowercase = None					,					lowercase = True					,					)     ->  Union[DDIMSchedulerOutput, Tuple]:
							'''simple docstring'''






							A__						=	timestep + self.config.num_train_timesteps // self.num_inference_steps

							# 2. compute alphas, betas
							# change original implementation to exactly match noise levels for analogous forward process
							A__						=	self.alphas_cumprod[timestep]
							A__						=	(
							    self.alphas_cumprod[prev_timestep]
							    if prev_timestep < self.config.num_train_timesteps
							    else self.final_alpha_cumprod
							)

							A__						=	1 - alpha_prod_t

							# 3. compute predicted original sample from predicted noise also called
							# "predicted x_0" of formula (12) from https://arxiv.org/pdf/2010.02502.pdf
							if self.config.prediction_type == "epsilon":
											A__						=	(sample - beta_prod_t ** 0.5 * model_output) / alpha_prod_t ** 0.5
											A__						=	model_output
							elif self.config.prediction_type == "sample":
											A__						=	model_output
											A__						=	(sample - alpha_prod_t ** 0.5 * pred_original_sample) / beta_prod_t ** 0.5
							elif self.config.prediction_type == "v_prediction":
											A__						=	(alpha_prod_t**0.5) * sample - (beta_prod_t**0.5) * model_output
											A__						=	(alpha_prod_t**0.5) * model_output + (beta_prod_t**0.5) * sample
							else:
											raise ValueError(
											    F'prediction_type given as {self.config.prediction_type} must be one of `epsilon`, `sample`, or'
											    " `v_prediction`"				)

							# 4. Clip or threshold "predicted x_0"
							if self.config.clip_sample:
											A__						=	pred_original_sample.clamp(
											    -self.config.clip_sample_range					,					self.config.clip_sample_range				)

							# 5. compute "direction pointing to x_t" of formula (12) from https://arxiv.org/pdf/2010.02502.pdf
							A__						=	(1 - alpha_prod_t_prev) ** 0.5 * pred_epsilon

							# 6. compute x_t without "random noise" of formula (12) from https://arxiv.org/pdf/2010.02502.pdf
							A__						=	alpha_prod_t_prev ** 0.5 * pred_original_sample + pred_sample_direction

							if not return_dict:
											return (prev_sample, pred_original_sample)
							return DDIMSchedulerOutput(prev_sample=lowercase					,					pred_original_sample=lowercase				)


			def __len__(			self				)     ->  Optional[Any]:
							'''simple docstring'''






							return self.config.num_train_timesteps





import unittest

from transformers import JukeboxTokenizer
from transformers.testing_utils import require_torch







class  snake_case__(							unittest.TestCase							):







						'''simple docstring'''




						SCREAMING_SNAKE_CASE__					:     Tuple    =							JukeboxTokenizer
						SCREAMING_SNAKE_CASE__					:     int    =							{
						    """artist""": """Zac Brown Band""",
						    """genres""": """Country""",
						    """lyrics""": """I met a traveller from an antique land,
        Who said \"Two vast and trunkless legs of stone
        Stand in the desert. . . . Near them, on the sand,
        Half sunk a shattered visage lies, whose frown,
        And wrinkled lip, and sneer of cold command,
        Tell that its sculptor well those passions read
        Which yet survive, stamped on these lifeless things,
        The hand that mocked them, and the heart that fed;
        And on the pedestal, these words appear:
        My name is Ozymandias, King of Kings;
        Look on my Works, ye Mighty, and despair!
        Nothing beside remains. Round the decay
        Of that colossal Wreck, boundless and bare
        The lone and level sands stretch far away
        """,
						}




						@require_torch
						def   lowercase_       ( self							)    ->			Union[str, Any]:
										import torch

										lowerCAmelCase_       :					Union[str, Any]			 = JukeboxTokenizer.from_pretrained('''openai/jukebox-1b-lyrics'''							)
										lowerCAmelCase_       :					Any			 = tokenizer(**self.metas							)['''input_ids''']
										# fmt: off
										lowerCAmelCase_       :					List[str]			 = [
										    torch.tensor([[
										        0, 0, 0, 7_1_6_9, 5_0_7, 9, 7_6, 3_9, 3_1, 4_6, 7_6, 2_7,
										        7_6, 4_6, 4_4, 2_7, 4_8, 3_1, 3_8, 3_8, 3_1, 4_4, 7_6, 3_2,
										        4_4, 4_1, 3_9, 7_6, 2_7, 4_0, 7_6, 2_7, 4_0, 4_6, 3_5, 4_3,
										        4_7, 3_1, 7_6, 3_8, 2_7, 4_0, 3_0, 6_4, 7_8, 7_6, 7_6, 7_6,
										        7_6, 7_6, 7_6, 7_6, 7_6, 2_3, 3_4, 4_1, 7_6, 4_5, 2_7, 3_5,
										        3_0, 7_6, 7_1, 2_0, 4_9, 4_1, 7_6, 4_8, 2_7, 4_5, 4_6, 7_6,
										        2_7, 4_0, 3_0, 7_6, 4_6, 4_4, 4_7, 4_0, 3_7, 3_8, 3_1, 4_5,
										        4_5, 7_6, 3_8, 3_1, 3_3, 4_5, 7_6, 4_1, 3_2, 7_6, 4_5, 4_6,
										        4_1, 4_0, 3_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
										        1_9, 4_6, 2_7, 4_0, 3_0, 7_6, 3_5, 4_0, 7_6, 4_6, 3_4, 3_1,
										        7_6, 3_0, 3_1, 4_5, 3_1, 4_4, 4_6, 6_3, 7_6, 6_3, 7_6, 6_3,
										        7_6, 6_3, 7_6, 1_4, 3_1, 2_7, 4_4, 7_6, 4_6, 3_4, 3_1, 3_9,
										        6_4, 7_6, 4_1, 4_0, 7_6, 4_6, 3_4, 3_1, 7_6, 4_5, 2_7, 4_0,
										        3_0, 6_4, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 8,
										        2_7, 3_8, 3_2, 7_6, 4_5, 4_7, 4_0, 3_7, 7_6, 2_7, 7_6, 4_5,
										        3_4, 2_7, 4_6, 4_6, 3_1, 4_4, 3_1, 3_0, 7_6, 4_8, 3_5, 4_5,
										        2_7, 3_3, 3_1, 7_6, 3_8, 3_5, 3_1, 4_5, 6_4, 7_6, 4_9, 3_4,
										        4_1, 4_5, 3_1, 7_6, 3_2, 4_4, 4_1, 4_9, 4_0, 6_4, 7_8, 7_6,
										        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1, 4_0, 3_0, 7_6, 4_9,
										        4_4, 3_5, 4_0, 3_7, 3_8, 3_1, 3_0, 7_6, 3_8, 3_5, 4_2, 6_4,
										        7_6, 2_7, 4_0, 3_0, 7_6, 4_5, 4_0, 3_1, 3_1, 4_4, 7_6, 4_1,
										        3_2, 7_6, 2_9, 4_1, 3_8, 3_0, 7_6, 2_9, 4_1, 3_9, 3_9, 2_7,
										        4_0, 3_0, 6_4, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
										        2_0, 3_1, 3_8, 3_8, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_5, 4_6,
										        4_5, 7_6, 4_5, 2_9, 4_7, 3_8, 4_2, 4_6, 4_1, 4_4, 7_6, 4_9,
										        3_1, 3_8, 3_8, 7_6, 4_6, 3_4, 4_1, 4_5, 3_1, 7_6, 4_2, 2_7,
										        4_5, 4_5, 3_5, 4_1, 4_0, 4_5, 7_6, 4_4, 3_1, 2_7, 3_0, 7_8,
										        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 2_3, 3_4, 3_5, 2_9,
										        3_4, 7_6, 5_1, 3_1, 4_6, 7_6, 4_5, 4_7, 4_4, 4_8, 3_5, 4_8,
										        3_1, 6_4, 7_6, 4_5, 4_6, 2_7, 3_9, 4_2, 3_1, 3_0, 7_6, 4_1,
										        4_0, 7_6, 4_6, 3_4, 3_1, 4_5, 3_1, 7_6, 3_8, 3_5, 3_2, 3_1,
										        3_8, 3_1, 4_5, 4_5, 7_6, 4_6, 3_4, 3_5, 4_0, 3_3, 4_5, 6_4,
										        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 2_0, 3_4, 3_1,
										        7_6, 3_4, 2_7, 4_0, 3_0, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_9,
										        4_1, 2_9, 3_7, 3_1, 3_0, 7_6, 4_6, 3_4, 3_1, 3_9, 6_4, 7_6,
										        2_7, 4_0, 3_0, 7_6, 4_6, 3_4, 3_1, 7_6, 3_4, 3_1, 2_7, 4_4,
										        4_6, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_2, 3_1, 3_0, 6_6, 7_8,
										        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1, 4_0, 3_0, 7_6,
										        4_1, 4_0, 7_6, 4_6, 3_4, 3_1, 7_6, 4_2, 3_1, 3_0, 3_1, 4_5,
										        4_6, 2_7, 3_8, 6_4, 7_6, 4_6, 3_4, 3_1, 4_5, 3_1, 7_6, 4_9,
										        4_1, 4_4, 3_0, 4_5, 7_6, 2_7, 4_2, 4_2, 3_1, 2_7, 4_4, 6_5,
										        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_3, 5_1, 7_6,
										        4_0, 2_7, 3_9, 3_1, 7_6, 3_5, 4_5, 7_6, 1_5, 5_2, 5_1, 3_9,
										        2_7, 4_0, 3_0, 3_5, 2_7, 4_5, 6_4, 7_6, 1_1, 3_5, 4_0, 3_3,
										        7_6, 4_1, 3_2, 7_6, 1_1, 3_5, 4_0, 3_3, 4_5, 6_6, 7_8, 7_6,
										        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_2, 4_1, 4_1, 3_7, 7_6,
										        4_1, 4_0, 7_6, 3_9, 5_1, 7_6, 2_3, 4_1, 4_4, 3_7, 4_5, 6_4,
										        7_6, 5_1, 3_1, 7_6, 1_3, 3_5, 3_3, 3_4, 4_6, 5_1, 6_4, 7_6,
										        2_7, 4_0, 3_0, 7_6, 3_0, 3_1, 4_5, 4_2, 2_7, 3_5, 4_4, 6_7,
										        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_4, 4_1, 4_6,
										        3_4, 3_5, 4_0, 3_3, 7_6, 2_8, 3_1, 4_5, 3_5, 3_0, 3_1, 7_6,
										        4_4, 3_1, 3_9, 2_7, 3_5, 4_0, 4_5, 6_3, 7_6, 1_8, 4_1, 4_7,
										        4_0, 3_0, 7_6, 4_6, 3_4, 3_1, 7_6, 3_0, 3_1, 2_9, 2_7, 5_1,
										        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_5, 3_2, 7_6,
										        4_6, 3_4, 2_7, 4_6, 7_6, 2_9, 4_1, 3_8, 4_1, 4_5, 4_5, 2_7,
										        3_8, 7_6, 2_3, 4_4, 3_1, 2_9, 3_7, 6_4, 7_6, 2_8, 4_1, 4_7,
										        4_0, 3_0, 3_8, 3_1, 4_5, 4_5, 7_6, 2_7, 4_0, 3_0, 7_6, 2_8,
										        2_7, 4_4, 3_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
										        2_0, 3_4, 3_1, 7_6, 3_8, 4_1, 4_0, 3_1, 7_6, 2_7, 4_0, 3_0,
										        7_6, 3_8, 3_1, 4_8, 3_1, 3_8, 7_6, 4_5, 2_7, 4_0, 3_0, 4_5,
										        7_6, 4_5, 4_6, 4_4, 3_1, 4_6, 2_9, 3_4, 7_6, 3_2, 2_7, 4_4,
										        7_6, 2_7, 4_9, 2_7, 5_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
										        7_6, 7_6]]							),
										    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1]]							),
										    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1]]							),
										]
										# fmt: on
										self.assertTrue(torch.allclose(tokens[0]      ,  EXPECTED_OUTPUT[0]							)							)
										self.assertTrue(torch.allclose(tokens[1]      ,  EXPECTED_OUTPUT[1]							)							)
										self.assertTrue(torch.allclose(tokens[2]      ,  EXPECTED_OUTPUT[2]							)							)




						@require_torch
						def   lowercase_       ( self							)    ->			List[Any]:
										import torch

										lowerCAmelCase_       :					Any			 = JukeboxTokenizer.from_pretrained('''openai/jukebox-5b-lyrics'''							)
										lowerCAmelCase_       :					str			 = tokenizer(**self.metas							)['''input_ids''']
										# fmt: off
										lowerCAmelCase_       :					Tuple			 = [
										    torch.tensor([[
										        0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1, 9, 7_7, 3_9,
										        3_1, 4_6, 7_7, 2_7, 7_7, 4_6, 4_4, 2_7, 4_8, 3_1, 3_8, 3_8,
										        3_1, 4_4, 7_7, 3_2, 4_4, 4_1, 3_9, 7_7, 2_7, 4_0, 7_7, 2_7,
										        4_0, 4_6, 3_5, 4_3, 4_7, 3_1, 7_7, 3_8, 2_7, 4_0, 3_0, 6_4,
										        7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 2_3, 3_4, 4_1,
										        7_7, 4_5, 2_7, 3_5, 3_0, 7_7, 7_2, 2_0, 4_9, 4_1, 7_7, 4_8,
										        2_7, 4_5, 4_6, 7_7, 2_7, 4_0, 3_0, 7_7, 4_6, 4_4, 4_7, 4_0,
										        3_7, 3_8, 3_1, 4_5, 4_5, 7_7, 3_8, 3_1, 3_3, 4_5, 7_7, 4_1,
										        3_2, 7_7, 4_5, 4_6, 4_1, 4_0, 3_1, 7_9, 7_7, 7_7, 7_7, 7_7,
										        7_7, 7_7, 7_7, 7_7, 1_9, 4_6, 2_7, 4_0, 3_0, 7_7, 3_5, 4_0,
										        7_7, 4_6, 3_4, 3_1, 7_7, 3_0, 3_1, 4_5, 3_1, 4_4, 4_6, 6_3,
										        7_7, 6_3, 7_7, 6_3, 7_7, 6_3, 7_7, 1_4, 3_1, 2_7, 4_4, 7_7,
										        4_6, 3_4, 3_1, 3_9, 6_4, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1,
										        7_7, 4_5, 2_7, 4_0, 3_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7,
										        7_7, 7_7, 7_7, 8, 2_7, 3_8, 3_2, 7_7, 4_5, 4_7, 4_0, 3_7,
										        7_7, 2_7, 7_7, 4_5, 3_4, 2_7, 4_6, 4_6, 3_1, 4_4, 3_1, 3_0,
										        7_7, 4_8, 3_5, 4_5, 2_7, 3_3, 3_1, 7_7, 3_8, 3_5, 3_1, 4_5,
										        6_4, 7_7, 4_9, 3_4, 4_1, 4_5, 3_1, 7_7, 3_2, 4_4, 4_1, 4_9,
										        4_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 1,
										        4_0, 3_0, 7_7, 4_9, 4_4, 3_5, 4_0, 3_7, 3_8, 3_1, 3_0, 7_7,
										        3_8, 3_5, 4_2, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 4_5, 4_0, 3_1,
										        3_1, 4_4, 7_7, 4_1, 3_2, 7_7, 2_9, 4_1, 3_8, 3_0, 7_7, 2_9,
										        4_1, 3_9, 3_9, 2_7, 4_0, 3_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7,
										        7_7, 7_7, 7_7, 7_7, 2_0, 3_1, 3_8, 3_8, 7_7, 4_6, 3_4, 2_7,
										        4_6, 7_7, 3_5, 4_6, 4_5, 7_7, 4_5, 2_9, 4_7, 3_8, 4_2, 4_6,
										        4_1, 4_4, 7_7, 4_9, 3_1, 3_8, 3_8, 7_7, 4_6, 3_4, 4_1, 4_5,
										        3_1, 7_7, 4_2, 2_7, 4_5, 4_5, 3_5, 4_1, 4_0, 4_5, 7_7, 4_4,
										        3_1, 2_7, 3_0, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        2_3, 3_4, 3_5, 2_9, 3_4, 7_7, 5_1, 3_1, 4_6, 7_7, 4_5, 4_7,
										        4_4, 4_8, 3_5, 4_8, 3_1, 6_4, 7_7, 4_5, 4_6, 2_7, 3_9, 4_2,
										        3_1, 3_0, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1, 4_5, 3_1, 7_7,
										        3_8, 3_5, 3_2, 3_1, 3_8, 3_1, 4_5, 4_5, 7_7, 4_6, 3_4, 3_5,
										        4_0, 3_3, 4_5, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        7_7, 2_0, 3_4, 3_1, 7_7, 3_4, 2_7, 4_0, 3_0, 7_7, 4_6, 3_4,
										        2_7, 4_6, 7_7, 3_9, 4_1, 2_9, 3_7, 3_1, 3_0, 7_7, 4_6, 3_4,
										        3_1, 3_9, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 4_6, 3_4, 3_1, 7_7,
										        3_4, 3_1, 2_7, 4_4, 4_6, 7_7, 4_6, 3_4, 2_7, 4_6, 7_7, 3_2,
										        3_1, 3_0, 6_6, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        1, 4_0, 3_0, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1, 7_7, 4_2,
										        3_1, 3_0, 3_1, 4_5, 4_6, 2_7, 3_8, 6_4, 7_7, 4_6, 3_4, 3_1,
										        4_5, 3_1, 7_7, 4_9, 4_1, 4_4, 3_0, 4_5, 7_7, 2_7, 4_2, 4_2,
										        3_1, 2_7, 4_4, 6_5, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        7_7, 1_3, 5_1, 7_7, 4_0, 2_7, 3_9, 3_1, 7_7, 3_5, 4_5, 7_7,
										        1_5, 5_2, 5_1, 3_9, 2_7, 4_0, 3_0, 3_5, 2_7, 4_5, 6_4, 7_7,
										        1_1, 3_5, 4_0, 3_3, 7_7, 4_1, 3_2, 7_7, 1_1, 3_5, 4_0, 3_3,
										        4_5, 6_6, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 1_2,
										        4_1, 4_1, 3_7, 7_7, 4_1, 4_0, 7_7, 3_9, 5_1, 7_7, 2_3, 4_1,
										        4_4, 3_7, 4_5, 6_4, 7_7, 5_1, 3_1, 7_7, 1_3, 3_5, 3_3, 3_4,
										        4_6, 5_1, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 3_0, 3_1, 4_5, 4_2,
										        2_7, 3_5, 4_4, 6_7, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        7_7, 1_4, 4_1, 4_6, 3_4, 3_5, 4_0, 3_3, 7_7, 2_8, 3_1, 4_5,
										        3_5, 3_0, 3_1, 7_7, 4_4, 3_1, 3_9, 2_7, 3_5, 4_0, 4_5, 6_3,
										        7_7, 1_8, 4_1, 4_7, 4_0, 3_0, 7_7, 4_6, 3_4, 3_1, 7_7, 3_0,
										        3_1, 2_9, 2_7, 5_1, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
										        7_7, 1_5, 3_2, 7_7, 4_6, 3_4, 2_7, 4_6, 7_7, 2_9, 4_1, 3_8,
										        4_1, 4_5, 4_5, 2_7, 3_8, 7_7, 2_3, 4_4, 3_1, 2_9, 3_7, 6_4,
										        7_7, 2_8, 4_1, 4_7, 4_0, 3_0, 3_8, 3_1, 4_5, 4_5, 7_7, 2_7,
										        4_0, 3_0, 7_7, 2_8, 2_7, 4_4, 3_1, 7_9, 7_7, 7_7, 7_7, 7_7,
										        7_7, 7_7, 7_7, 7_7, 2_0, 3_4, 3_1, 7_7, 3_8, 4_1, 4_0, 3_1,
										        7_7, 2_7, 4_0, 3_0, 7_7, 3_8, 3_1, 4_8, 3_1, 3_8, 7_7, 4_5,
										        2_7, 4_0, 3_0, 4_5, 7_7, 4_5, 4_6, 4_4, 3_1, 4_6, 2_9, 3_4,
										        7_7, 3_2, 2_7, 4_4, 7_7, 2_7, 4_9, 2_7, 5_1, 7_9, 7_7, 7_7,
										        7_7, 7_7, 7_7, 7_7, 7_7, 7_7]]							),
										    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1]]							),
										    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1]]							),
										]
										# fmt: on
										self.assertTrue(torch.allclose(tokens[0]      ,  EXPECTED_OUTPUT[0]							)							)
										self.assertTrue(torch.allclose(tokens[1]      ,  EXPECTED_OUTPUT[1]							)							)
										self.assertTrue(torch.allclose(tokens[2]      ,  EXPECTED_OUTPUT[2]							)							)