bglick13's picture
Update model_index.json
d87a910
raw
history blame contribute delete
No virus
339 Bytes
{
"_class_name": "ValueGuidedRLPipeline",
"_diffusers_version": "0.6.0",
"scheduler": [
"diffusers",
"DDPMScheduler"
],
"unet": [
"diffusers",
"UNet1DModel"
],
"value_function": [
"diffusers",
"UNet1DModel"
],
"args": ["diffusers", "DDPMScheduler"],
"kwargs": ["diffusers", "PNDMScheduler"]
}