Upload IDArbDiffusionPipeline

model_index.json

@@ -1,7 +1,7 @@
 {
-  "_class_name": "DRDiffusionImagePipeline",
+  "_class_name": "IDArbDiffusionPipeline",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "outputs/irmv-v2-1/pipeckpts",
+  "_name_or_path": "./pipeckpts",
   "feature_extractor": [
     "transformers",
     "CLIPImageProcessor"
@@ -13,7 +13,7 @@
   ],
   "scheduler": [
     "diffusers",
-    "DDIMScheduler"
+    "EulerAncestralDiscreteScheduler"
   ],
   "text_encoder": [
     "transformers",
@@ -24,7 +24,7 @@
     "CLIPTokenizer"
   ],
   "unet": [
-    "irmvdiffusion.models.unet_dr2d_condition",
+    "idarbdiffusion.models.unet_dr2d_condition",
     "UNetDR2DConditionModel"
   ],
   "vae": [

scheduler/scheduler_config.json

@@ -1,20 +1,16 @@
 {
-  "_class_name": "DDIMScheduler",
+  "_class_name": "EulerAncestralDiscreteScheduler",
   "_diffusers_version": "0.31.0",
   "beta_end": 0.012,
   "beta_schedule": "linear",
   "beta_start": 0.00085,
   "clip_sample": false,
-  "clip_sample_range": 1.0,
-  "dynamic_thresholding_ratio": 0.995,
   "num_train_timesteps": 1000,
   "prediction_type": "v_prediction",
   "rescale_betas_zero_snr": false,
-  "sample_max_value": 1.0,
   "set_alpha_to_one": false,
   "skip_prk_steps": true,
   "steps_offset": 1,
-  "thresholding": false,
   "timestep_spacing": "linspace",
   "trained_betas": null
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "outputs/irmv-v2-1/pipeckpts/text_encoder",
+  "_name_or_path": "./pipeckpts/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNetDR2DConditionModel",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "outputs/irmv-v2-1/pipeckpts/unet",
+  "_name_or_path": "./pipeckpts/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "outputs/irmv-v2-1/pipeckpts/vae",
+  "_name_or_path": "./pipeckpts/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,