Spaces:

miccull
/

clip-text-to-rgb

Runtime error

miccull commited on Oct 29, 2021

Commit

9ca1ed9

•

1 Parent(s): 3c27e5d

no more cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ def gradio_fn(text_prompt, adam_learning_rate, adam_weight_decay, n_iterations=5
   opt = torch.optim.AdamW([rgb_model()], lr=adam_learning_rate, weight_decay=adam_weight_decay)
   with torch.no_grad():
-    tokenized_text = clip.tokenize(text_prompt).cuda()
     target_embedding = model.encode_text(tokenized_text).detach().clone()
   def training_step():

   opt = torch.optim.AdamW([rgb_model()], lr=adam_learning_rate, weight_decay=adam_weight_decay)
   with torch.no_grad():
+    tokenized_text = clip.tokenize(text_prompt).to(DEVICE)
     target_embedding = model.encode_text(tokenized_text).detach().clone()
   def training_step():