Hugging Face's logo

Spaces:
miccull
/
clip-rgb-interpolation
Runtime error

App Files Community
miccull commited on
Commit
0d248fe
·
1 Parent(s): 4602ab6

no more cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def encode_color(color):
29
  return model.encode_image( resizer(rgb) )
30
 
31
  def encode_text(text):
32
- tokenized_text = clip.tokenize(text).cuda()
33
  return model.encode_text(tokenized_text)
34
 
35
  def lerp(x, y, steps=11):
 
29
  return model.encode_image( resizer(rgb) )
30
 
31
  def encode_text(text):
32
+ tokenized_text = clip.tokenize(text).to(DEVICE)
33
  return model.encode_text(tokenized_text)
34
 
35
  def lerp(x, y, steps=11):