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))
|