



'''simple docstring'''

from arguments import InitializationArguments

from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, HfArgumentParser


# Configuration
__lowerCAmelCase						      =       HfArgumentParser(InitializationArguments)
__lowerCAmelCase						      =       parser.parse_args()

# Load codeparrot tokenizer trained for Python code tokenization
__lowerCAmelCase						      =       AutoTokenizer.from_pretrained(args.tokenizer_name)

# Config: "scale_attn_by_layer_idx" and "reorder_and_upcast_attn" are Mistral stability tweaks
__lowerCAmelCase						      =       {
    '''vocab_size''': len(tokenizer),
    '''scale_attn_by_inverse_layer_idx''': True,
    '''reorder_and_upcast_attn''': True,
}

# Load model config (GPT-2 large in this case)
__lowerCAmelCase						      =       AutoConfig.from_pretrained(args.config_name, **config_kwargs)

# Initialize new model with config
__lowerCAmelCase						      =       AutoModelForCausalLM.from_config(config)

# Save model to the hub
model.save_pretrained(args.model_name, push_to_hub=args.push_to_hub)









'''simple docstring'''

from __future__ import annotations

import os
from typing import Any

import requests

__lowerCAmelCase						      =       '''https://api.github.com'''

# https://docs.github.com/en/free-pro-team@latest/rest/reference/users#get-the-authenticated-user
__lowerCAmelCase						      =       BASE_URL + '''/user'''

# https://github.com/settings/tokens
__lowerCAmelCase						      =       os.environ.get('''USER_TOKEN''', '''''')



def 					__lowerCamelCase					(				lowerCAmelCase_  )      ->				dict[Any, Any]:
		_a					:				Union[str, Any]				=							{
		    'Authorization': f"""token {auth_token}""",
		    'Accept': 'application/vnd.github.v3+json',
		}
		return requests.get(lowerCAmelCase_						,      headers=lowerCAmelCase_  ).json()


if __name__ == "__main__":  # pragma: no cover
							if USER_TOKEN:
														for key, value in fetch_github_info(USER_TOKEN).items():
																					print(f"""{key}: {value}""")
							else:
														raise ValueError('''\'USER_TOKEN\' field cannot be empty.''')




