Mou11209203
commited on
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -90,7 +90,7 @@ scheduler = DDPMScheduler.from_pretrained(repo_id1)
|
|
90 |
print("scheduler.config: ", scheduler.config)
|
91 |
|
92 |
# 3. Create an image with Gaussian noise
|
93 |
-
torch.manual_seed(
|
94 |
noisy_sample = torch.randn(1, model.config.in_channels, model.config.sample_size, model.config.sample_size).to("cuda")
|
95 |
print(f"Noisy sample shape: {noisy_sample.shape}")
|
96 |
|
|
|
90 |
print("scheduler.config: ", scheduler.config)
|
91 |
|
92 |
# 3. Create an image with Gaussian noise
|
93 |
+
torch.manual_seed(1733783271) # Fix the random seed for reproducibility
|
94 |
noisy_sample = torch.randn(1, model.config.in_channels, model.config.sample_size, model.config.sample_size).to("cuda")
|
95 |
print(f"Noisy sample shape: {noisy_sample.shape}")
|
96 |
|