Update model.py

model.py

@@ -44,7 +44,7 @@ class Model:
         print("image tokenizer is loaded")
         return vq_model
 
-    def load_gpt(self, condition_type='canny'):
+    def load_gpt(self, condition_type='edge'):
         gpt_ckpt = models[condition_type]
         # precision = torch.bfloat16
         precision = torch.float32
@@ -57,7 +57,7 @@ class Model:
             adapter_size='base',
         ).to(device='cpu', dtype=precision)
         model_weight = load_file(gpt_ckpt)
-        gpt_model.load_state_dict(model_weight, strict=True)
+        gpt_model.load_state_dict(model_weight, strict=False)
         gpt_model.eval()
         print("gpt model is loaded")
         return gpt_model