Update README.md

README.md

@@ -13,4 +13,25 @@ If you create a fun image with this model, please show your result and [@FeiArt_
 Or you can join the [FeiArt Diffusion Discord](https://discord.gg/MkAsEpNnqs)
 Share your work created with this model.
 Exchange experiences and parameters.
-And see more custom diffusion models
+And see more custom diffusion models
+
+
+model_config.update({
+          'attention_resolutions': '32,16,8',
+          'class_cond': False,
+          'diffusion_steps': 1000,
+          'rescale_timesteps': True,
+          'timestep_respacing': 'ddim100',
+          'image_size': 512,
+          'learn_sigma': True,
+          'noise_schedule': 'linear',
+          'num_channels': 256,
+          'num_head_channels': 64,
+          'num_res_blocks': 2,
+          'resblock_updown': True,
+          'use_checkpoint': use_checkpoint,
+          'use_fp16': True,
+          'use_scale_shift_norm': True,
+          
+          and change the model.load to below
+          model.load_state_dict(torch.load(custom_path, map_location='cpu'), strict=False)