Unverified Commit 1221b28e authored by Alphin Jain's avatar Alphin Jain Committed by GitHub
Browse files

Fix AttributeError in train_lcm_distill_lora_sdxl_wds.py (#7923)



Fix conditional teacher model check in train_lcm_distill_lora_sdxl_wds.py
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 746f603b
......@@ -1358,7 +1358,7 @@ def main(args):
# estimates to predict the data point in the augmented PF-ODE trajectory corresponding to the next ODE
# solver timestep.
with torch.no_grad():
if torch.backends.mps.is_available() or "playground" in args.pretrained_model_name_or_path:
if torch.backends.mps.is_available() or "playground" in args.pretrained_teacher_model:
autocast_ctx = nullcontext()
else:
autocast_ctx = torch.autocast(accelerator.device.type)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment