fastAPI_CatVTON / info.txt
nrtoya's picture
0135
ea56a4f
raw
history blame contribute delete
682 Bytes
-- 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 ๋ถ€๋ถ„ ์ˆ˜์ •