U
    &c                     @   s   d dl mZmZ d dlZedZedZedZe	egZ
dZedD ]Zee
ed\ZZed ddf Zed dj Zd	d
 ej D Zed d Zdd
 eD Zee g7 Zed Z
eee g qNeeZee dS )    )GPT2LMHeadModelGPT2TokenizerNZgpt2zThe Manhattan bridge   )past
   c                 C   s   g | ]}t |gqS  )	tokenizerdecode).0xr   r   6/tmp/pip-unpacked-wheel-ymerj3tt/transformers/filep.py
<listcomp>   s     r      c                 C   s   g | ]}t |qS r   )r	   encode)r   wordr   r   r   r      s     )Ztransformersr   r   ZtorchZfrom_pretrainedr	   modelr   	generatedZtensorcontextr   rangeioutputdistributionZtopkindicestolistZ
top_valuesZ	top_wordswordsZselected_tokensZargmax_tokenZ	unsqueezeprintr
   sequencer   r   r   r   <module>   s$   




