|
-- gpu ํ๊ฒฝ : nvidia rtx 2080 |
|
-- ์ ์ ์์์ hugging face repository clone ํ ์์ ์ฌํญ |
|
- torch==2.0.1 torchvision==0.15.2 ์ผ๋ก ๋ฎ์ ๋ฒ์ ์ค์นํ์ |
|
- default="booksforcharlie/stable-diffusion-inpainting" ์ด ๋ถ๋ถ์์, .safetensors ํ์ผ ์์ด์ pickle๋ก .bin ํ์ผ ์๋์ค์น๋จ |
|
- app.py์์ line 81. mixed precision default ๊ฐ no ๋ก ์ค์ |
|
- pipeline.py ์์ line53๋ถ๋ถ ์ฃผ์์ฒ๋ฆฌ |
|
|
|
|
|
-- ์ดํ masking ์์ฑ๊ณผ ๊ด๋ จํ ๋ถ๋ถ customizing์ ์ํด app.py ์์ |
|
- fitting_type ๋งค๊ฐ๋ณ์ submit_function() ๋ฉ์๋ ๋ฑ์ ์ถ๊ฐํ์์.(๊ทธ ์ธ ๊ด๋ จ๋ ๋ถ๋ถ๋ค์) |
|
- app.py์ # process mask ๋ถ๋ถ ์์ |