ethanNeuralImage commited on
Commit
ac43402
1 Parent(s): 6208cab

CLIP score use CPU please

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -211,7 +211,7 @@ with gr.Blocks() as demo:
211
 
212
  return lpips_score, ssim_score, id_score
213
  def clip_text_metric(tensor, text):
214
- clip_embed = torch.cat([clip.tokenize(text)]).cuda()
215
  clip_score = 1-clip_text(tensor.unsqueeze(0), clip_embed).item()
216
  return clip_score
217
 
 
211
 
212
  return lpips_score, ssim_score, id_score
213
  def clip_text_metric(tensor, text):
214
+ clip_embed = torch.cat([clip.tokenize(text)]).cpu()
215
  clip_score = 1-clip_text(tensor.unsqueeze(0), clip_embed).item()
216
  return clip_score
217