PommesPeter commited on
Commit
4c702d6
1 Parent(s): cc5ce56

Update models/__init__.py

Browse files
Files changed (1) hide show
  1. models/__init__.py +1 -2
models/__init__.py CHANGED
@@ -1,2 +1 @@
1
- # from .model import DiT_Llama_5B_patch2
2
- from .model import DiT_Llama_2B_patch2
 
1
+ from .model import NextDiT_2B_patch2