Update README.md
Browse files
README.md
CHANGED
@@ -13,4 +13,25 @@ If you create a fun image with this model, please show your result and [@FeiArt_
|
|
13 |
Or you can join the [FeiArt Diffusion Discord](https://discord.gg/MkAsEpNnqs)
|
14 |
Share your work created with this model.
|
15 |
Exchange experiences and parameters.
|
16 |
-
And see more custom diffusion models
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
Or you can join the [FeiArt Diffusion Discord](https://discord.gg/MkAsEpNnqs)
|
14 |
Share your work created with this model.
|
15 |
Exchange experiences and parameters.
|
16 |
+
And see more custom diffusion models
|
17 |
+
|
18 |
+
|
19 |
+
model_config.update({
|
20 |
+
'attention_resolutions': '32,16,8',
|
21 |
+
'class_cond': False,
|
22 |
+
'diffusion_steps': 1000,
|
23 |
+
'rescale_timesteps': True,
|
24 |
+
'timestep_respacing': 'ddim100',
|
25 |
+
'image_size': 512,
|
26 |
+
'learn_sigma': True,
|
27 |
+
'noise_schedule': 'linear',
|
28 |
+
'num_channels': 256,
|
29 |
+
'num_head_channels': 64,
|
30 |
+
'num_res_blocks': 2,
|
31 |
+
'resblock_updown': True,
|
32 |
+
'use_checkpoint': use_checkpoint,
|
33 |
+
'use_fp16': True,
|
34 |
+
'use_scale_shift_norm': True,
|
35 |
+
|
36 |
+
and change the model.load to below
|
37 |
+
model.load_state_dict(torch.load(custom_path, map_location='cpu'), strict=False)
|