add fp32 for cpu

app.py

@@ -82,6 +82,7 @@ def range_loss(input):
     
 def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed, image_prompts,timestep_respacing, cutn):
     # Model settings
+    device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
     model_config = model_and_diffusion_defaults()
     model_config.update({
           'attention_resolutions': '16',
@@ -96,12 +97,12 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
           'num_heads': 1,
           'num_res_blocks': 2,
           'use_checkpoint': True,
-          'use_fp16': True,
+          'use_fp16': False if device == 'cpu' else True,
           'use_scale_shift_norm': False,
       })
 
     # Load models
-    device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+    
     print('Using device:', device)
     model, diffusion = create_model_and_diffusion(**model_config)
     model.load_state_dict(torch.load('256x256_openai_comics_faces_by_alex_spirin_084000.pt', map_location='cpu'))