Smile_Changer / models

Commit History

Fix CUDA operations to check device: only use fused ops when input is on CUDA
80d11e4

LogicGoInfotechSpaces commited on

Add error handling for missing checkpoint files in load_weights methods
bfac43d

LogicGoInfotechSpaces commited on

Add device detection and CPU fallback directly in FSEFull and FSEInverter constructors
372a1fe

LogicGoInfotechSpaces commited on

Fix device handling in FSEFull and FSEInverter: convert device string to torch.device object
87d55be

LogicGoInfotechSpaces commited on

Add error handling for missing StyleGAN discriminator weights in both FSEFull and FSEInverter classes
51cbd8f

LogicGoInfotechSpaces commited on

Add error handling for missing ArcFace model in PSP encoder
b885690

LogicGoInfotechSpaces commited on

CPU fallback: avoid CUDA JIT by disabling fused ops when build unavailable; force CPU
704a5f0

LogicGoInfotechSpaces commited on

Bundle StyleFeatureEditor code packages in Space to fix ModuleNotFoundError
95b1715

LogicGoInfotechSpaces commited on