Update nodes.py

This commit is contained in:
kijai
2024-02-10 14:59:05 +02:00
parent 2911a53a9a
commit 158fac4aa3
+1 -1
View File
@@ -752,7 +752,7 @@ class TrainMotionDirectorLora:
unet.to(device)
vae.to(device)
text_encoder.to(device)
tokenizer.to(device)
target_spatial_modules = ["Transformer3DModel"]
target_temporal_modules = ["TemporalTransformerBlock"]