Skip to content

Commit

Permalink
Merge pull request #63 from pozitronik/fix_model_uri
Browse files Browse the repository at this point in the history
Fix model uri
  • Loading branch information
pozitronik authored Aug 24, 2023
2 parents a6527d8 + 5aff841 commit e234ebd
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 3 deletions.
1 change: 0 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,6 @@ sinner uses the same ML libraries to perform its magic, but handles them in its
## Credits

- [s0md3v](https://github.com/s0md3v/): the original author of roop
- [henryruhs](https://github.com/henryruhs): the significant contributor to roop
- [ffmpeg](https://ffmpeg.org/): for making video related operations easy
- [deepinsight](https://github.com/deepinsight): for their [insightface](https://github.com/deepinsight/insightface) project which provided a well-made library and models.
- [ARC Lab, Tencent PCG](https://github.com/TencentARC): for their [GFPGAN](https://github.com/TencentARC/GFPGAN) project which provided a face restoration library and models.
Expand Down
2 changes: 1 addition & 1 deletion sinner/processors/frame/FaceEnhancer.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ def face_enhancer(self) -> GFPGANer:

def __init__(self, parameters: Namespace, target_path: str | None = None) -> None:
download_directory_path = get_app_dir('models')
conditional_download(download_directory_path, ['https://huggingface.co/henryruhs/roop/resolve/main/GFPGANv1.4.pth'])
conditional_download(download_directory_path, ['https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth'])
super().__init__(parameters, target_path)

def enhance_face(self, temp_frame: Frame) -> Frame:
Expand Down
2 changes: 1 addition & 1 deletion sinner/processors/frame/FaceSwapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def face_swapper(self) -> FaceSwapperType:

def __init__(self, parameters: Namespace, target_path: str | None = None) -> None:
download_directory_path = get_app_dir('models')
conditional_download(download_directory_path, ['https://huggingface.co/henryruhs/roop/resolve/main/inswapper_128.onnx'])
conditional_download(download_directory_path, ['https://github.com/pozitronik/sinner/releases/download/v200823/inswapper_128.onnx'])
super().__init__(parameters, target_path)

def process_frame(self, frame: Frame) -> Frame:
Expand Down

0 comments on commit e234ebd

Please sign in to comment.