Spaces:

Gradio-Blocks
/

clip-guided-faces

Runtime error

sxela commited on May 28, 2022

Commit

c3cf91d

1 Parent(s): f9507c6

add device fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,7 +97,7 @@ 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': False if device == 'cpu' else True,
           'use_scale_shift_norm': False,
       })

           'num_heads': 1,
           'num_res_blocks': 2,
           'use_checkpoint': True,
+          'use_fp16': False if device.type == 'cpu' else True,
           'use_scale_shift_norm': False,
       })