Spaces:

ucalyptus
/

PTI

Runtime error

PTI / configs /paths_config.py

trial

06229a5 over 2 years ago

No virus

991 Bytes

	## Pretrained models paths
	e4e = 'e4e_ffhq_encode.pt'
	stylegan2_ada_ffhq = 'ffhq.pkl'
	style_clip_pretrained_mappers = ''
	ir_se50 = 'model_ir_se50.pth'
	dlib = 'align.dat'

	## Dirs for output files
	checkpoints_dir = './checkpoints'
	embedding_base_dir = './embeddings'
	styleclip_output_dir = './StyleCLIP_results'
	experiments_output_dir = './output'

	## Input info
	### Input dir, where the images reside
	input_data_path = ''
	### Inversion identifier, used to keeping track of the inversion results. Both the latent code and the generator
	input_data_id = 'rocky'

	## Keywords
	pti_results_keyword = 'PTI'
	e4e_results_keyword = 'e4e'
	sg2_results_keyword = 'SG2'
	sg2_plus_results_keyword = 'SG2_plus'
	multi_id_model_type = 'multi_id'

	## Edit directions
	interfacegan_age = 'editings/interfacegan_directions/age.pt'
	interfacegan_smile = 'editings/interfacegan_directions/smile.pt'
	interfacegan_rotation = 'editings/interfacegan_directions/rotation.pt'
	ffhq_pca = 'editings/ganspace_pca/ffhq_pca.pt'