This commit is contained in:
matt3o
2023-12-02 08:20:20 +01:00
parent 701f54d977
commit 50bde71f23
-1
View File
@@ -217,7 +217,6 @@ class CrossAttentionPatch:
self.unconds = [uncond]
self.dtype = dtype
self.device = 'cuda' if 'cuda' in device.type else 'cpu'
print(self.device)
self.number = number
self.weight_type = [weight_type]
self.masks = [mask]