Hugging Face's logo

Yuanzhi
/
DMD2_humble

Model card Files Community
Yuanzhi commited on
Commit
b2609c2
·
verified ·
1 Parent(s): d2f23e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -3,7 +3,11 @@ license: apache-2.0
3
  ---
4
 
5
  Home-made DMD2 on laion recap dataset.
 
6
  rename the safetensor to diffusion_pytorch_model.safetensors
 
7
  load the unet with unet = UNet2DConditionModel.from_pretrained(current_path).to("cuda", torch.float16)
 
8
  current_path is the path to the safetensors and config.json
9
- The rest is the same as official DMD2
 
 
3
  ---
4
 
5
  Home-made DMD2 on laion recap dataset.
6
+
7
  rename the safetensor to diffusion_pytorch_model.safetensors
8
+
9
  load the unet with unet = UNet2DConditionModel.from_pretrained(current_path).to("cuda", torch.float16)
10
+
11
  current_path is the path to the safetensors and config.json
12
+
13
+ The rest is the same as official DMD2 https://huggingface.co/tianweiy/DMD2