Spaces:
Runtime error
Runtime error
add fp16 log
Browse files
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'))
|