changes to torch dtype and CPU support only

app.py

@@ -4,6 +4,7 @@ from geo_painting import GeoPainting
 
 def generate_image(input_promp, control_image):
     new_geo_painting = GeoPainting()
+    print("::::::::::generate_image:::::::::::::")
     print(type(control_image))
     return new_geo_painting.generate_painting(input_promp, control_image)
 

geo_painting.py

@@ -13,18 +13,18 @@ class GeoPainting:
     DEFAULT_DIFFUSER_MODEL = "geospatial_diffuser"
 
     def __init__(self, controlnet_model_path=DEFAULT_CONTROLNET_MODEL, diffuser_model=DEFAULT_DIFFUSER_MODEL):
-        self.controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float16)
+        self.controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float32)
         self.generator = torch.Generator(device="cpu").manual_seed(2)
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
             diffuser_model,
             low_cpu_mem_usage=False,
             device_map=None,
             controlnet=self.controlnet,
-            torch_dtype=torch.float16
+            torch_dtype=torch.float32
         )
         self.pipe.scheduler = UniPCMultistepScheduler.from_config(self.pipe.scheduler.config)
-        self.pipe.enable_model_cpu_offload()
         if torch.cuda.is_available():
+            self.pipe.enable_model_cpu_offload()
             self.pipe.enable_xformers_memory_efficient_attention()
 
     def generate_painting(self, input_promp, control_image):