multimodalart HF staff commited on
Commit
7af4a09
1 Parent(s): eecb1f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ make_cutouts = MakeCutouts(clip_model.visual.input_resolution, 16, 1.)
59
  def run_all(prompt, steps, n_images, weight, clip_guided):
60
  import random
61
  seed = int(random.randint(0, 2147483647))
62
- target_embed = clip_model.encode_text(clip.tokenize(prompt)).float()#.cuda()
63
 
64
  if(clip_guided):
65
  prompts = [prompt]
 
59
  def run_all(prompt, steps, n_images, weight, clip_guided):
60
  import random
61
  seed = int(random.randint(0, 2147483647))
62
+ target_embed = clip_model.encode_text(clip.tokenize(prompt).to('cuda')).float()#.cuda()
63
 
64
  if(clip_guided):
65
  prompts = [prompt]