Spaces:

yfyangd
/

diffusion

Runtime error

yfyangd commited on Sep 4, 2022

Commit

1583096

1 Parent(s): 8bf6c54

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
         if 'qkv' in name or 'norm' in name or 'proj' in name:
             param.requires_grad_()
     if model_config['use_fp16']:
-        model.convert_to_fp32()
     clip_model = clip.load('ViT-B/16', jit=False)[0].eval().requires_grad_(False).to(device)
     clip_size = clip_model.visual.input_resolution
     normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],

         if 'qkv' in name or 'norm' in name or 'proj' in name:
             param.requires_grad_()
     if model_config['use_fp16']:
+        model.convert_to_fp16()
     clip_model = clip.load('ViT-B/16', jit=False)[0].eval().requires_grad_(False).to(device)
     clip_size = clip_model.visual.input_resolution
     normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],