lschmidt commited on
Commit
aab5a27
·
verified ·
1 Parent(s): 87d6e87

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +46 -0
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - super-resolution
5
+ - ddim
6
+ - weather
7
+ - wind
8
+ library_name: diffusers
9
+ model_type: ddim
10
+ datasets:
11
+ - your-dataset-name
12
+ ---
13
+ # DDIM-DSC (4× Super-Resolution for Wind Data)
14
+ ---
15
+
16
+ ## 🧠 Model Architecture
17
+
18
+ - **Base**: DDIM
19
+ - **Input channels**: 10
20
+ - **Output channels**: 10
21
+ - **Scale factor**: 4×
22
+ ---
23
+
24
+ ---
25
+ ## 🚀 How to Use
26
+ ```python
27
+ import torch
28
+ from diffusers import DiffusionPipeline, DDIMScheduler
29
+ from diffusers import DDIMScheduler
30
+
31
+ #
32
+ pipe = DiffusionPipeline.from_pretrained(
33
+ "lschmidt/ddim-dsc",
34
+ custom_pipeline="cond_ddim_pipeline",
35
+ trust_remote_code=True
36
+ )
37
+
38
+ #
39
+ batch_size = 1
40
+ in_channels = 6
41
+ sample_size = 160
42
+ sequence_len = 3
43
+
44
+ lres_image = torch.randn((sequence_len, in_channels, sample_size, sample_size)).to(pipe.device)
45
+ print(inputs.shape)
46
+ pipe(image =inputs)