silentchen commited on
Commit
aa8d774
1 Parent(s): f30c271

update space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -286,7 +286,7 @@ def main():
286
  os.makedirs(general_save_path, exist_ok=True)
287
  for i, latent in enumerate(state['latent']):
288
  latent = latent.to(device)
289
- text_embeddings_clip = model.cached_model_kwargs(1, dict(texts=[instruction])).to(device)
290
  print("shape of latent: ", latent.clone().unsqueeze(0).shape, "instruction: ", instruction)
291
  ref_latent = latent.clone().unsqueeze(0).to(device)
292
  t_1 = torch.randint(noise_start_t_e_type, noise_start_t_e_type + 1, (1,), device=device).long()
 
286
  os.makedirs(general_save_path, exist_ok=True)
287
  for i, latent in enumerate(state['latent']):
288
  latent = latent.to(device)
289
+ text_embeddings_clip = model.cached_model_kwargs(1, dict(texts=[instruction]))
290
  print("shape of latent: ", latent.clone().unsqueeze(0).shape, "instruction: ", instruction)
291
  ref_latent = latent.clone().unsqueeze(0).to(device)
292
  t_1 = torch.randint(noise_start_t_e_type, noise_start_t_e_type + 1, (1,), device=device).long()