Spaces:
Configuration error
Configuration error
change image save dir
Browse files- 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))
|