qwen-image splited training

This commit is contained in:
Artiprocher
2025-09-02 16:44:14 +08:00
parent 260e32217f
commit b6da77e468
7 changed files with 221 additions and 14 deletions

6
test.py Normal file
View File

@@ -0,0 +1,6 @@
import torch
data = torch.load("models/train/Qwen-Image_lora_cache/0/0.pth", map_location="cpu", weights_only=False)
for i in data:
print(i)