Harley-ml
/

MNIST-IMG-390k

digit_diffusion

Model card Files Files and versions

Harley-ml commited on 7 days ago

Commit

b9aba18

·

verified ·

1 Parent(s): 837c003

Update inference.py

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -60,7 +60,7 @@ def generate_grid(
     num_inference_steps: int,
     image_size: int,
 ) -> torch.Tensor:
-    """Generate a single grid image containing all requested digits."""
     scheduler.set_timesteps(num_inference_steps, device=device)
     rows: list[torch.Tensor] = []

     num_inference_steps: int,
     image_size: int,
 ) -> torch.Tensor:
+    # Generate a single grid image containing all requested digits.
     scheduler.set_timesteps(num_inference_steps, device=device)
     rows: list[torch.Tensor] = []