correct configs

model_index.json

@@ -1,6 +1,6 @@
 {
-  "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.7.2",
+  "_class_name": "AltDiffusionPipeline",
+  "_diffusers_version": "0.8.0.dev0",
   "feature_extractor": [
     "transformers",
     "CLIPFeatureExtractor"
@@ -11,11 +11,11 @@
   ],
   "scheduler": [
     "diffusers",
-    "DDIMScheduler"
+    "PNDMScheduler"
   ],
   "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
+    "alt_diffusion",
+    "RobertaSeriesModelWithTransformation"
   ],
   "tokenizer": [
     "transformers",

safety_checker/config.json

@@ -79,7 +79,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0",
+    "transformers_version": "4.24.0.dev0",
     "typical_p": 1.0,
     "use_bfloat16": false,
     "vocab_size": 49408
@@ -165,7 +165,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0",
+    "transformers_version": "4.24.0.dev0",
     "typical_p": 1.0,
     "use_bfloat16": false
   },

scheduler/scheduler_config.json

@@ -1,12 +1,13 @@
 {
-  "_class_name": "DDIMScheduler",
-  "_diffusers_version": "0.7.2",
+  "_class_name": "PNDMScheduler",
+  "_diffusers_version": "0.6.0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
-  "clip_sample": false,
   "num_train_timesteps": 1000,
   "set_alpha_to_one": false,
+  "skip_prk_steps": true,
   "steps_offset": 1,
-  "trained_betas": null
+  "trained_betas": null,
+  "clip_sample": false
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "xlm-roberta-large",
+  "_name_or_path": "../test/text_encoder",
   "architectures": [
     "RobertaSeriesModelWithTransformation"
   ],
@@ -24,8 +24,9 @@
   "position_embedding_type": "absolute",
   "project_dim": 768,
   "torch_dtype": "float32",
-  "transformers_version": "4.24.0",
+  "transformers_version": "4.24.0.dev0",
   "type_vocab_size": 1,
+  "use_attention_mask": true,
   "use_cache": true,
   "vocab_size": 250002
 }

tokenizer/tokenizer_config.json

@@ -10,10 +10,9 @@
     "rstrip": false,
     "single_word": false
   },
-  "model_max_length": 512,
-  "name_or_path": "/sharefs/baai-mrnd/yzd/test/xlm-roberta-large",
+  "model_max_length": 77,
+  "name_or_path": "../test/tokenizer",
   "pad_token": "<pad>",
-  "processor_class": "CHCLIPProcess",
   "sep_token": "</s>",
   "sp_model_kwargs": {},
   "special_tokens_map_file": null,

unet/config.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.7.2",
+  "_diffusers_version": "0.8.0.dev0",
+  "_name_or_path": "../test/unet",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

vae/config.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.7.2",
+  "_diffusers_version": "0.8.0.dev0",
+  "_name_or_path": "../test/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,