🚚 (docs) split by modules

docs/make.jl

@@ -16,5 +16,7 @@ makedocs(;
   ),
   pages=[
     "Home" => "index.md",
+    "Schedulers" => "schedulers.md",
+    "Beta Schedules" => "beta_schedules.md",
   ]
 )

docs/src/beta_schedules.md

@@ -0,0 +1,32 @@
+```@eval
+using Diffusers.BetaSchedules
+using Plots
+plotlyjs()
+
+T = 1000
+linear = linear_beta_schedule(T)
+scaled_linear = scaled_linear_beta_schedule(T)
+cosine = cosine_beta_schedule(T)
+sigmoid = sigmoid_beta_schedule(T)
+
+plot(
+  [linear, scaled_linear, cosine, sigmoid],
+  label=["linear" "scaled_linear" "cosine" "sigmoid"],
+  xlabel="t",
+  ylabel="β",
+  title="Beta schedules",
+  legend=:topleft,
+  yscale=:log10,
+)
+
+savefig("fig_beta_schedules.html")
+nothing
+```
+
+```@raw html
+<object type="text/html" data="fig_beta_schedules.html" style="width:100%;height:420px;"></object>
+```
+
+```@autodocs
+Modules = [Diffusers.BetaSchedules]
+```

docs/src/index.md

@@ -6,46 +6,7 @@ CurrentModule = Diffusers
 
 Documentation for [Diffusers.jl](https://github.com/Laurent2916/Diffusers.jl).
 
-```@eval
-using Diffusers.BetaSchedules
-using Plots
-plotlyjs()
-
-T = 1000
-linear = linear_beta_schedule(T)
-scaled_linear = scaled_linear_beta_schedule(T)
-cosine = cosine_beta_schedule(T)
-sigmoid = sigmoid_beta_schedule(T)
-
-plot(
-  [linear, scaled_linear, cosine, sigmoid],
-  label=["linear" "scaled_linear" "cosine" "sigmoid"],
-  xlabel="t",
-  ylabel="β",
-  title="Beta schedules",
-  legend=:topleft,
-  yscale=:log10,
-)
-
-savefig("beta_schedules.html")
-nothing
-```
-
-```@raw html
-<object type="text/html" data="beta_schedules.html" style="width:100%;height:420px;"></object>
-```
+## Index
 
 ```@index
 ```
-
-```@autodocs
-Modules = [Diffusers]
-```
-
-```@autodocs
-Modules = [Diffusers.BetaSchedules]
-```
-
-```@autodocs
-Modules = [Diffusers.Schedulers]
-```

docs/src/schedulers.md

@@ -0,0 +1,3 @@
+```@autodocs
+Modules = [Diffusers.Schedulers]
+```