From 290ec469cae943b91f0b8691a178b6f316413ea1 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Tue, 6 May 2025 17:54:32 +0800 Subject: [PATCH] train --- train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train.py b/train.py index aaa11e7..5aa81d8 100644 --- a/train.py +++ b/train.py @@ -254,7 +254,6 @@ class UnifiedModel(pl.LightningModule): self.pipe.vae_decoder.requires_grad_(False) self.pipe.vae_encoder.requires_grad_(False) self.pipe.text_encoder_1.requires_grad_(False) - self.qwenvl.requires_grad_(False) self.pipe.scheduler.set_timesteps(1000, training=True)