miccull commited on
Commit
9ca1ed9
1 Parent(s): 3c27e5d

no more cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def gradio_fn(text_prompt, adam_learning_rate, adam_weight_decay, n_iterations=5
57
  opt = torch.optim.AdamW([rgb_model()], lr=adam_learning_rate, weight_decay=adam_weight_decay)
58
 
59
  with torch.no_grad():
60
- tokenized_text = clip.tokenize(text_prompt).cuda()
61
  target_embedding = model.encode_text(tokenized_text).detach().clone()
62
 
63
  def training_step():
 
57
  opt = torch.optim.AdamW([rgb_model()], lr=adam_learning_rate, weight_decay=adam_weight_decay)
58
 
59
  with torch.no_grad():
60
+ tokenized_text = clip.tokenize(text_prompt).to(DEVICE)
61
  target_embedding = model.encode_text(tokenized_text).detach().clone()
62
 
63
  def training_step():