





import doctest

import projectq
from projectq.ops import H, Measure


def        __snake_case       (						lowerCAmelCase_  )							->	int:
    SCREAMING_SNAKE_CASE__       							= quantum_engine.allocate_qubit()
    H | qubit
    Measure | qubit
    return int(lowerCAmelCase_  )


if __name__ == "__main__":
  doctest.testmod()

  # initialises a new quantum backend
  _A				:							Tuple							= projectq.MainEngine()

  # Generate a list of 10 random numbers
  _A				:							str							= [get_random_number(quantum_engine) for _ in range(10)]

  # Flushes the quantum engine from memory
  quantum_engine.flush()

  print("""Random numbers""", random_numbers_list)







import os
from pathlib import Path
from typing import Dict, List

import fire
import torch

from transformers import AutoModelForSeqaSeqLM, AutoTokenizer
from transformers.utils.logging import get_logger


_A				:							Any							= get_logger(__name__)


def        __snake_case       (						lowerCAmelCase_   ,       lowerCAmelCase_  )							->	str:
    if text.startswith(lowerCAmelCase_  ):
        return text[len(lowerCAmelCase_  ) :]
    return text  # or whatever


def        __snake_case       (						lowerCAmelCase_  )							->	Union[str, Any]:
    return {remove_prefix(lowerCAmelCase_   ,       '''model.'''  ): v for k, v in sd.items()}


def        __snake_case       (						lowerCAmelCase_  )							->	Optional[int]:
    SCREAMING_SNAKE_CASE__       							= {}
    for k in state_dicts[0].keys():
        SCREAMING_SNAKE_CASE__       							= [sd[k] for sd in state_dicts]
        SCREAMING_SNAKE_CASE__       							= sum(lowerCAmelCase_  ) / len(lowerCAmelCase_  )
        assert isinstance(lowerCAmelCase_   ,       torch.Tensor  )
        SCREAMING_SNAKE_CASE__       							= new_t
    return new_sd


def        __snake_case       (						lowerCAmelCase_   ,       lowerCAmelCase_   ,       lowerCAmelCase_  )							->	None:
    SCREAMING_SNAKE_CASE__       							= AutoModelForSeqaSeqLM.from_pretrained(lowerCAmelCase_  )
    if os.path.isfile(lowerCAmelCase_  ):
        SCREAMING_SNAKE_CASE__       							= [pl_ckpt_path]
    else:
        assert os.path.isdir(lowerCAmelCase_  )
        SCREAMING_SNAKE_CASE__       							= list(Path(lowerCAmelCase_  ).glob('''*.ckpt'''  )  )
        assert ckpt_files, f'''could not find any ckpt files inside the {pl_ckpt_path} directory'''

    if len(lowerCAmelCase_  ) > 1:
        logger.info(f'''averaging the weights of {ckpt_files}'''  )

    SCREAMING_SNAKE_CASE__       							= [sanitize(torch.load(lowerCAmelCase_   ,       map_location='''cpu'''  )['''state_dict''']  ) for x in ckpt_files]
    SCREAMING_SNAKE_CASE__       							= average_state_dicts(lowerCAmelCase_  )

    SCREAMING_SNAKE_CASE__							,		SCREAMING_SNAKE_CASE__       							= hf_model.load_state_dict(lowerCAmelCase_   ,       strict=lowerCAmelCase_  )
    assert not missing, f'''missing keys: {missing}'''
    hf_model.save_pretrained(lowerCAmelCase_  )
    try:
        SCREAMING_SNAKE_CASE__       							= AutoTokenizer.from_pretrained(lowerCAmelCase_  )
        tok.save_pretrained(lowerCAmelCase_  )
    except Exception:
        pass
        # dont copy tokenizer if cant


if __name__ == "__main__":
  fire.Fire(convert_pl_to_hf)
