Update app.py
Browse files
app.py
CHANGED
@@ -107,7 +107,7 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
|
|
107 |
if 'qkv' in name or 'norm' in name or 'proj' in name:
|
108 |
param.requires_grad_()
|
109 |
if model_config['use_fp16']:
|
110 |
-
model.
|
111 |
clip_model = clip.load('ViT-B/16', jit=False)[0].eval().requires_grad_(False).to(device)
|
112 |
clip_size = clip_model.visual.input_resolution
|
113 |
normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],
|
|
|
107 |
if 'qkv' in name or 'norm' in name or 'proj' in name:
|
108 |
param.requires_grad_()
|
109 |
if model_config['use_fp16']:
|
110 |
+
model.convert_to_fp16()
|
111 |
clip_model = clip.load('ViT-B/16', jit=False)[0].eval().requires_grad_(False).to(device)
|
112 |
clip_size = clip_model.visual.input_resolution
|
113 |
normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],
|