Spaces:
Runtime error
Runtime error
ethanNeuralImage
commited on
Commit
•
ac43402
1
Parent(s):
6208cab
CLIP score use CPU please
Browse files
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)]).
|
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 |
|