Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -100,7 +100,8 @@ with open(scheduler_file_path, 'r') as f:
|
|
| 100 |
scheduler_config = json.load(f)
|
| 101 |
|
| 102 |
noise_scheduler = DDPMScheduler.from_config(scheduler_config)
|
| 103 |
-
```
|
|
|
|
| 104 |
|
| 105 |
Use the model to generate images:
|
| 106 |
|
|
@@ -129,4 +130,3 @@ for i, t in tqdm(enumerate(noise_scheduler.timesteps)):
|
|
| 129 |
fig, ax = plt.subplots(1, 1, figsize=(12, 12))
|
| 130 |
ax.imshow(torchvision.utils.make_grid(x.detach().cpu().clip(-1, 1), nrow=8)[0], cmap='Greys')
|
| 131 |
```
|
| 132 |
-
|
|
|
|
| 100 |
scheduler_config = json.load(f)
|
| 101 |
|
| 102 |
noise_scheduler = DDPMScheduler.from_config(scheduler_config)
|
| 103 |
+
```
|
| 104 |
+
|
| 105 |
|
| 106 |
Use the model to generate images:
|
| 107 |
|
|
|
|
| 130 |
fig, ax = plt.subplots(1, 1, figsize=(12, 12))
|
| 131 |
ax.imshow(torchvision.utils.make_grid(x.detach().cpu().clip(-1, 1), nrow=8)[0], cmap='Greys')
|
| 132 |
```
|
|
|