multimodalart HF staff commited on
Commit
d8035da
1 Parent(s): bfdd665

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def run_all(prompt, steps, n_images, weight, clip_guided):
33
  import random
34
  seed = int(random.randint(0, 2147483647))
35
  target_embed = clip_model.encode_text(clip.tokenize(prompt)).float().cuda()
36
- clip_embed = target_embed.repeat([n, 1])
37
  def cfg_model_fn(x, t):
38
  """The CFG wrapper function."""
39
  n = x.shape[0]
 
33
  import random
34
  seed = int(random.randint(0, 2147483647))
35
  target_embed = clip_model.encode_text(clip.tokenize(prompt)).float().cuda()
36
+ clip_embed = target_embed.repeat([n_images, 1])
37
  def cfg_model_fn(x, t):
38
  """The CFG wrapper function."""
39
  n = x.shape[0]