Open Dong-Ki-Lee opened 2 years ago
When loading with the previous code, both pl_sd itself and the model read by state_dict from pl_sd are stored in vram, so the program consumes more graphics memory than it needs. Modify that part.
When loading with the previous code, both pl_sd itself and the model read by state_dict from pl_sd are stored in vram, so the program consumes more graphics memory than it needs. Modify that part.