diff --git a/nodes.py b/nodes.py index bed0c04..edd90db 100644 --- a/nodes.py +++ b/nodes.py @@ -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"]