Srikumar26 commited on
Commit
93dff8a
1 Parent(s): 6b6d14e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -6
README.md CHANGED
@@ -31,7 +31,7 @@ This is a ControlNet based model that synthesizes satellite images given OpenStr
31
  ## Examples
32
 
33
  ```python
34
- from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
35
  import torch
36
  from PIL import Image
37
 
@@ -39,11 +39,8 @@ img = Image.open("osm_tile_18_42048_101323.jpeg")
39
 
40
  controlnet = ControlNetModel.from_pretrained("MVRL/GeoSynth-OSM")
41
 
42
- scheduler = UniPCMultistepScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base", subfolder="scheduler")
43
-
44
- pipe = StableDiffusionControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", controlnet=controlnet, scheduler=scheduler)
45
- pipe.enable_xformers_memory_efficient_attention()
46
- pipe.enable_model_cpu_offload()
47
 
48
  # generate image
49
  generator = torch.manual_seed(10345340)
 
31
  ## Examples
32
 
33
  ```python
34
+ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
35
  import torch
36
  from PIL import Image
37
 
 
39
 
40
  controlnet = ControlNetModel.from_pretrained("MVRL/GeoSynth-OSM")
41
 
42
+ pipe = StableDiffusionControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", controlnet=controlnet)
43
+ pipe = pipe.to("cuda:0")
 
 
 
44
 
45
  # generate image
46
  generator = torch.manual_seed(10345340)