Wolfvin commited on
Commit
9c1a00d
·
verified ·
1 Parent(s): d158200

Upload diffusion_llm/model/__init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. diffusion_llm/model/__init__.py +5 -0
diffusion_llm/model/__init__.py CHANGED
@@ -4,10 +4,15 @@ from diffusion_llm.model.noise_scheduler import NoiseScheduler
4
  from diffusion_llm.model.graph_encoder import GraphConditioningEncoder
5
  from diffusion_llm.model.diffusion_transformer import DiffusionTransformer
6
  from diffusion_llm.model.aam_diffusion_model import AamDiffusionModel
 
 
7
 
8
  __all__ = [
9
  "NoiseScheduler",
10
  "GraphConditioningEncoder",
11
  "DiffusionTransformer",
12
  "AamDiffusionModel",
 
 
 
13
  ]
 
4
  from diffusion_llm.model.graph_encoder import GraphConditioningEncoder
5
  from diffusion_llm.model.diffusion_transformer import DiffusionTransformer
6
  from diffusion_llm.model.aam_diffusion_model import AamDiffusionModel
7
+ from diffusion_llm.model.mirror_speculative import MirrorSpeculativeDecoder, MirrorSpeculativeConfig
8
+ from diffusion_llm.model.evoformer import RouterExpertCoevolve
9
 
10
  __all__ = [
11
  "NoiseScheduler",
12
  "GraphConditioningEncoder",
13
  "DiffusionTransformer",
14
  "AamDiffusionModel",
15
+ "MirrorSpeculativeDecoder",
16
+ "MirrorSpeculativeConfig",
17
+ "RouterExpertCoevolve",
18
  ]