intel / ai-reference-models

Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Intel® Data Center GPUs
Apache License 2.0
683 stars 220 forks source link

update SD finetune tests #172

Closed jiqing-feng closed 8 months ago

jiqing-feng commented 8 months ago

Hi @WafaaT @jianan-gu . This PR updates the textual inversion and also extracts loss to verify if the training process is successful.

cc @DiweiSun