prs-eth
/

marigold-normals-lcm-v0-1

MarigoldPipeline

monocular normals estimation

single image normals estimation

normals-estimation

Model card Files Files and versions Community

toshas commited on Apr 27, 2024

Commit

c967abf

·

1 Parent(s): 3dc099e

fix LCMScheduler configuration

Files changed (2) hide show

model_index.json +17 -18
scheduler/scheduler_config.json +20 -19

model_index.json CHANGED Viewed

@@ -1,25 +1,24 @@
 {
-  "_class_name": "MarigoldNormalsPipeline",
-  "_diffusers_version": "0.27.2",
-  "_name_or_path": "prs-eth/marigold-normals-lcm-v0-1",
-  "scheduler": [
-    "diffusers",
-    "DDIMScheduler"
   ],
-  "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
   ],
-  "tokenizer": [
-    "transformers",
-    "CLIPTokenizer"
   ],
-  "unet": [
-    "diffusers",
-    "UNet2DConditionModel"
   ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
   ]
 }

 {
+  "_class_name":"MarigoldPipeline",
+  "_diffusers_version":"0.25.0",
+  "unet":[
+     "diffusers",
+     "UNet2DConditionModel"
   ],
+  "vae":[
+     "diffusers",
+     "AutoencoderKL"
   ],
+  "scheduler":[
+     "diffusers",
+     "LCMScheduler"
   ],
+  "text_encoder":[
+     "transformers",
+     "CLIPTextModel"
   ],
+  "tokenizer":[
+     "transformers",
+     "CLIPTokenizer"
   ]
 }

scheduler/scheduler_config.json CHANGED Viewed

@@ -1,20 +1,21 @@
 {
-    "_class_name": "LCMScheduler",
-    "_diffusers_version": "0.22.0.dev0",
-    "beta_end": 0.012,
-    "beta_schedule": "scaled_linear",
-    "beta_start": 0.00085,
-    "clip_sample": false,
-    "clip_sample_range": 1.0,
-    "dynamic_thresholding_ratio": 0.995,
-    "num_train_timesteps": 1000,
-    "original_inference_steps": 50,
-    "prediction_type": "epsilon",
-    "rescale_betas_zero_snr": false,
-    "sample_max_value": 1.0,
-    "set_alpha_to_one": true,
-    "steps_offset": 1,
-    "thresholding": false,
-    "timestep_spacing": "leading",
-    "trained_betas": null
-  }

 {
+  "_class_name": "LCMScheduler",
+  "_diffusers_version": "0.25.0",
+  "beta_end": 0.012,
+  "beta_schedule": "scaled_linear",
+  "beta_start": 0.00085,
+  "clip_sample": false,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "num_train_timesteps": 1000,
+  "original_inference_steps": 50,
+  "prediction_type": "v_prediction",
+  "rescale_betas_zero_snr": false,
+  "sample_max_value": 1.0,
+  "set_alpha_to_one": false,
+  "steps_offset": 1,
+  "thresholding": false,
+  "timestep_scaling": 10.0,
+  "timestep_spacing": "leading",
+  "trained_betas": null
+}