misterruffian commited on
Commit
b0d131a
·
1 Parent(s): dbadd6e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
README.md CHANGED
@@ -1,3 +1,34 @@
1
  ---
2
  license: cc-by-4.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-4.0
3
  ---
4
+
5
+ Install Instructions
6
+ 1. Download Model into Google Drive > AI > DiscoDiffusion > Models
7
+ 2. Add path '/content/drive/MyDrive/AI/DiscoDiffusion/Models/AIDM_130k_v01.pt' to Disco Diffusion Step 2 > Custom Model > Custom Path
8
+ 3. In Custom Model Settings add the following code below
9
+ 4. Run All
10
+
11
+ Custom Model Settings
12
+ ---
13
+
14
+ #@markdown ####**Custom Model Settings:**
15
+ if diffusion_model == 'custom':
16
+ model_config.update({
17
+ 'attention_resolutions': '32, 16, 8',
18
+ 'class_cond': False,
19
+ 'diffusion_steps': 1000,
20
+ 'rescale_timesteps': True,
21
+ 'image_size': 512,
22
+ 'learn_sigma': True,
23
+ 'noise_schedule': 'linear',
24
+ 'num_channels': 128,
25
+ 'num_heads': 4,
26
+ 'num_res_blocks': 2,
27
+ 'resblock_updown': True,
28
+ 'use_checkpoint': use_checkpoint,
29
+ 'use_fp16': True,
30
+ 'use_scale_shift_norm': True,
31
+ })
32
+
33
+
34
+