From 158fac4aa3d7518f2ccaed86765d9060e4d605a2 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 10 Feb 2024 14:59:05 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"]