ddim scheduler for the upscaler

model_index.json

@@ -1,9 +1,9 @@
 {
   "_class_name": "GLIDE",
-  "_diffusers_version": "0.0.1",
-  "_module": "modeling_glide.py",
+  "_diffusers_version": "0.0.3",
+  "_module": "pipeline_glide",
   "text_encoder": [
-    "modeling_glide",
+    "diffusers",
     "CLIPTextModel"
   ],
   "text_noise_scheduler": [
@@ -20,7 +20,7 @@
   ],
   "upscale_noise_scheduler": [
     "diffusers",
-    "GlideDDIMScheduler"
+    "DDIMScheduler"
   ],
   "upscale_unet": [
     "diffusers",

text_encoder/config.json

@@ -18,7 +18,7 @@
   "num_hidden_layers": 16,
   "pad_token_id": 1,
   "torch_dtype": "float32",
-  "transformers_version": "4.20.0.dev0",
+  "transformers_version": "4.19.4",
   "use_padding_embeddings": true,
   "vocab_size": 50257
 }

text_noise_scheduler/scheduler_config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "ClassifierFreeGuidanceScheduler",
-  "_diffusers_version": "0.0.1",
+  "_diffusers_version": "0.0.3",
   "beta_schedule": "squaredcos_cap_v2",
   "timesteps": 1000
 }

text_unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "GLIDETextToImageUNetModel",
-  "_diffusers_version": "0.0.1",
+  "_diffusers_version": "0.0.3",
   "attention_resolutions": [
     2,
     4,
@@ -23,6 +23,7 @@
   "num_res_blocks": 3,
   "out_channels": 6,
   "resblock_updown": true,
+  "resolution": 64,
   "transformer_dim": 512,
   "use_checkpoint": false,
   "use_fp16": false,

tokenizer/special_tokens_map.json

@@ -1,30 +1 @@
-{
-  "bos_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "eos_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "pad_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "unk_token": {
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}
+{"bos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "pad_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}}

tokenizer/tokenizer_config.json

@@ -1,38 +1 @@
-{
-  "add_bos_token": false,
-  "add_prefix_space": false,
-  "bos_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "eos_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "errors": "replace",
-  "pad_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  },
-  "tokenizer_class": "GPT2Tokenizer",
-  "unk_token": {
-    "__type": "AddedToken",
-    "content": "<|endoftext|>",
-    "lstrip": false,
-    "normalized": true,
-    "rstrip": false,
-    "single_word": false
-  }
-}
+{"errors": "replace", "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "bos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "pad_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "add_prefix_space": false, "add_bos_token": false, "tokenizer_class": "GPT2Tokenizer"}

upscale_noise_scheduler/scheduler_config.json

@@ -1,6 +1,8 @@
 {
-  "_class_name": "GlideDDIMScheduler",
-  "_diffusers_version": "0.0.1",
+  "_class_name": "DDIMScheduler",
+  "_diffusers_version": "0.0.3",
+  "beta_end": 0.02,
   "beta_schedule": "linear",
+  "beta_start": 0.0001,
   "timesteps": 1000
 }

upscale_unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "GLIDESuperResUNetModel",
-  "_diffusers_version": "0.0.1",
+  "_diffusers_version": "0.0.3",
   "attention_resolutions": [
     8,
     16,
@@ -25,6 +25,7 @@
   "num_res_blocks": 2,
   "out_channels": 6,
   "resblock_updown": true,
+  "resolution": 256,
   "use_checkpoint": false,
   "use_fp16": false,
   "use_scale_shift_norm": true