Add diffusers model_index.json and scheduler config for DFlashPipeline.from_pretrained
#1
by kashif HF Staff - opened
- model_index.json +9 -0
- scheduler/scheduler_config.json +4 -0
model_index.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "DFlashPipeline",
|
| 3 |
+
"_diffusers_version": "0.40.0.dev0",
|
| 4 |
+
"_target_model_name_or_path": "z-lab/Alpamayo-R1-10B",
|
| 5 |
+
"scheduler": [
|
| 6 |
+
"diffusers",
|
| 7 |
+
"DFlashTokenDiffusionScheduler"
|
| 8 |
+
]
|
| 9 |
+
}
|
scheduler/scheduler_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "DFlashTokenDiffusionScheduler",
|
| 3 |
+
"_diffusers_version": "0.40.0.dev0"
|
| 4 |
+
}
|