From 3a83ec17318dc60ed46b4a3279d3dcbe7e8b02de Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 23 Dec 2022 23:02:01 +0100 Subject: Better dataset prompt handling --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index e272b5d..cc208f0 100644 --- a/train_ti.py +++ b/train_ti.py @@ -621,7 +621,7 @@ def main(): ).to(accelerator.device) pipeline.set_progress_bar_config(dynamic_ncols=True) - with torch.autocast("cuda"), torch.inference_mode(): + with torch.inference_mode(): for batch in batched_data: image_name = [item.class_image_path for item in batch] prompt = [item.cprompt for item in batch] -- cgit v1.2.3-54-g00ecf