erwann commited on
Commit
9251ae0
1 Parent(s): f2e0f04

change image save dir

Browse files
Files changed (1) hide show
  1. ImageState.py +3 -2
ImageState.py CHANGED
@@ -23,6 +23,7 @@ from img_processing import *
23
  from img_processing import custom_to_pil
24
  from loaders import load_default
25
  num = 0
 
26
  class PromptTransformHistory():
27
  def __init__(self, iterations) -> None:
28
  self.iterations = iterations
@@ -115,7 +116,7 @@ class ImageState:
115
  global num
116
  # global vqgan
117
  if self.state_id is None:
118
- self.state_id = str(uuid.uuid4())
119
  print("redner all", self.state_id)
120
  current_vector_transforms = (self.blue_eyes, self.lip_size, self.hair_gp, self.asian_transform, sum(self.current_prompt_transforms))
121
  new_latent = self.blend_latent + sum(current_vector_transforms)
@@ -174,7 +175,7 @@ class ImageState:
174
  return self._render_all_transformations()
175
  def apply_prompts(self, positive_prompts, negative_prompts, lr, iterations, lpips_weight, reconstruction_steps):
176
  if self.state_id is None:
177
- self.state_id = "./" + str(uuid.uuid4())
178
  transform_log = PromptTransformHistory(iterations + reconstruction_steps)
179
  transform_log.transforms.append(torch.zeros_like(self.blend_latent, requires_grad=False))
180
  self.current_prompt_transforms.append(torch.zeros_like(self.blend_latent, requires_grad=False))
 
23
  from img_processing import custom_to_pil
24
  from loaders import load_default
25
  num = 0
26
+
27
  class PromptTransformHistory():
28
  def __init__(self, iterations) -> None:
29
  self.iterations = iterations
 
116
  global num
117
  # global vqgan
118
  if self.state_id is None:
119
+ self.state_id = "./img_history/" + str(uuid.uuid4())
120
  print("redner all", self.state_id)
121
  current_vector_transforms = (self.blue_eyes, self.lip_size, self.hair_gp, self.asian_transform, sum(self.current_prompt_transforms))
122
  new_latent = self.blend_latent + sum(current_vector_transforms)
 
175
  return self._render_all_transformations()
176
  def apply_prompts(self, positive_prompts, negative_prompts, lr, iterations, lpips_weight, reconstruction_steps):
177
  if self.state_id is None:
178
+ self.state_id = "./img_history/" + str(uuid.uuid4())
179
  transform_log = PromptTransformHistory(iterations + reconstruction_steps)
180
  transform_log.transforms.append(torch.zeros_like(self.blend_latent, requires_grad=False))
181
  self.current_prompt_transforms.append(torch.zeros_like(self.blend_latent, requires_grad=False))