remove last cuda call

2024-07-10 04:00:03 +03:00
parent 07a68ab38d
commit 8ebd13c930
1 changed files with 1 additions and 1 deletions
@@ -296,7 +296,7 @@ class KolorsTextEncode:
                max_length=max_length,
                truncation=True,
                return_tensors="pt",
-            ).to('cuda')
+            ).to(device)
            output = text_encoder(
                    input_ids=uncond_input['input_ids'] ,
                    attention_mask=uncond_input['attention_mask'],