Hugging Face's logo

Spaces:
chenyangqi
/
FateZero
Runtime error

App Files Community
2
chenyangqi commited on
Commit
a06cc14
·
1 Parent(s): c4333c6

pretrained_model_path

Browse files
Files changed (1) hide show
  1. inference_fatezero.py +2 -1
inference_fatezero.py CHANGED
@@ -22,7 +22,8 @@ def merge_config_then_run(
22
 
23
  dataset_time_string = get_time_string()
24
  config_now = copy.deepcopy(Omegadict_default_edit_config)
25
- # config_now['pretrained_model_path'] = model_id
 
26
  config_now['train_dataset']['prompt'] = source_prompt
27
  config_now['train_dataset']['path'] = data_path
28
  config_now['validation_sample_logger_config']['prompts'] = [target_prompt]
 
22
 
23
  dataset_time_string = get_time_string()
24
  config_now = copy.deepcopy(Omegadict_default_edit_config)
25
+ print(f"config_now['pretrained_model_path'] = model_id {model_id}")
26
+ config_now['pretrained_model_path'] = model_id
27
  config_now['train_dataset']['prompt'] = source_prompt
28
  config_now['train_dataset']['path'] = data_path
29
  config_now['validation_sample_logger_config']['prompts'] = [target_prompt]