in_channels

hlky · hlky · commit 46a3b43d7cc5 · 2025-03-10T13:18:51.000Z
diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py
@@ -467,7 +467,7 @@ def _test_encode(self, **kwargs):
         width = kwargs.get("width")
 
         tensor = torch.randn(
-            (batch, self.model.config.latent_channels, height, width), dtype=self.dtype, device="cuda"
+            (batch, self.model.config.in_channels, height, width), dtype=self.dtype, device="cuda"
         )
 
         try:

Original file line number	Diff line number	Diff line change
`@@ -467,7 +467,7 @@ def _test_encode(self, **kwargs):`
`467`	`467`	`width = kwargs.get("width")`
`468`	`468`
`469`	`469`	`tensor = torch.randn(`
`470`		`- (batch, self.model.config.latent_channels, height, width), dtype=self.dtype, device="cuda"`
	`470`	`+ (batch, self.model.config.in_channels, height, width), dtype=self.dtype, device="cuda"`
`471`	`471`	`)`
`472`	`472`
`473`	`473`	`try:`