Adding FProcess catch for discrete flows, and readme update

Author: murrellb

README.md

@@ -13,11 +13,12 @@ The animated logo shows samples from a model trained to jointly transport a 2D p
 
 ## Features
 
-- Controllable noise (or fully deterministic for flow matching)
 - Flexible initial $X_0$ distribution
 - Conditioning via masking
 - States: Continuous, discrete, and a wide variety of manifolds supported (via [Manifolds.jl](https://github.com/JuliaManifolds/Manifolds.jl))
 - Compound states supported (e.g. jointly sampling from both continuous and discrete variables)
+- Controllable noise (or fully deterministic for flow matching)
+- Time-scaling schedules (see `examples/logo_example.jl`)
 
 ### Basic idea:
 - Generate `X0` and `X1` states from your favorite distribution, and a random `t` between 0 and 1

src/processes.jl

@@ -2,6 +2,13 @@
 #For processes that aren't used elsewhere
 ##########################################
 
+##########################################
+#https://arxiv.org/pdf/2407.15595
+##########################################
+
+#Note to future self: if I ever use the probability velocity trick for one of the regular discrete CTMCs (thus turning it into a ConvexInterpolatingDiscreteFlow), I'll need to handle the FProcess schedule a little differently.
+FProcess(p::ConvexInterpolatingDiscreteFlow, f) = Error("ConvexInterpolatingDiscreteFlow have their own schedule mechanisms. Do not use them with FProcess.")
+
 """
     InterpolatingDiscreteFlow(κ::Function, κ̇::Function)
     InterpolatingDiscreteFlow() - Uses default Cosine scheduler.

src/types.jl

@@ -50,7 +50,7 @@ Adapt.adapt_structure(to, G::Guide) = Guide(Adapt.adapt(to, G.H), Adapt.adapt(to
 UState = Union{State,MaskedState, Guide}
 
 #This is for all Flow types where the mixture probabilities are directly defined, and the gen is done via probability velocities.
-abstract type ConvexInterpolatingDiscreteFlow <: DiscreteProcess end
+abstract type ConvexInterpolatingDiscreteFlow <: DiscreteProcess end #https://arxiv.org/pdf/2407.15595
 
 struct InterpolatingDiscreteFlow <: ConvexInterpolatingDiscreteFlow
     κ::Function