CVL-Heidelberg
/

ControlNet-XS

Model card Files Files and versions Community

Sipirius commited on Sep 22, 2023

Commit

53a9b86

·

1 Parent(s): 0fe8c05

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ import torch
 from PIL import Image
 path_to_config = 'ControlNet-XS-main/configs/inference/sdxl/sdxl_encD_canny_48m.yaml'
-model = cu.create_model(path_to_config)
 image_path = 'PATH/TO/IMAGES/Shoe.png'
@@ -63,7 +63,7 @@ import torch
 from PIL import Image
 path_to_config = 'PATH/TO/CONFIG/sd21_encD_depth_14m.yaml'
-model = cu.create_model(path_to_config)
 size = 768
 image_path = 'PATH/TO/IMAGES/Shoe.png'

 from PIL import Image
 path_to_config = 'ControlNet-XS-main/configs/inference/sdxl/sdxl_encD_canny_48m.yaml'
+model = cu.create_model(path_to_config).to('cuda')
 image_path = 'PATH/TO/IMAGES/Shoe.png'
 from PIL import Image
 path_to_config = 'PATH/TO/CONFIG/sd21_encD_depth_14m.yaml'
+model = cu.create_model(path_to_config).to('cuda')
 size = 768
 image_path = 'PATH/TO/IMAGES/Shoe.png'