levihsu commited on
Commit
98ecf8d
1 Parent(s): 04cc394

Update preprocess/humanparsing/run_parsing.py

Browse files
preprocess/humanparsing/run_parsing.py CHANGED
@@ -22,8 +22,8 @@ class Parsing:
22
  session_options.add_session_config_entry('gpu_id', str(gpu_id))
23
 
24
  parsing_ckpt_path = os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing')
25
- atr_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/blob/main/checkpoints/humanparsing/parsing_atr.onnx'
26
- lip_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/blob/main/checkpoints/humanparsing/parsing_lip.onnx'
27
 
28
  from basicsr.utils.download_util import load_file_from_url
29
  load_file_from_url(atr_model_path, model_dir=parsing_ckpt_path)
 
22
  session_options.add_session_config_entry('gpu_id', str(gpu_id))
23
 
24
  parsing_ckpt_path = os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing')
25
+ atr_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/resolve/main/checkpoints/humanparsing/parsing_atr.onnx'
26
+ lip_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/resolve/main/checkpoints/humanparsing/parsing_lip.onnx'
27
 
28
  from basicsr.utils.download_util import load_file_from_url
29
  load_file_from_url(atr_model_path, model_dir=parsing_ckpt_path)