sxela commited on
Commit
a6da7af
Β·
1 Parent(s): f9507c6

add fp16 log

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
102
  })
103
 
104
  # Load models
105
-
106
  print('Using device:', device)
107
  model, diffusion = create_model_and_diffusion(**model_config)
108
  model.load_state_dict(torch.load('256x256_openai_comics_faces_by_alex_spirin_084000.pt', map_location='cpu'))
 
102
  })
103
 
104
  # Load models
105
+ print('Using fp16: ',model_config['use_fp16'])
106
  print('Using device:', device)
107
  model, diffusion = create_model_and_diffusion(**model_config)
108
  model.load_state_dict(torch.load('256x256_openai_comics_faces_by_alex_spirin_084000.pt', map_location='cpu'))