Update Era_s20_updt.py
Browse files- Era_s20_updt.py +2 -2
Era_s20_updt.py
CHANGED
@@ -193,7 +193,7 @@ def get_output_embeds(input_embeddings):
|
|
193 |
def generate_with_embs_custom(text_embeddings,seed):
|
194 |
height = 512 # default height of Stable Diffusion
|
195 |
width = 512 # default width of Stable Diffusion
|
196 |
-
num_inference_steps =
|
197 |
guidance_scale = 7.5 # Scale for classifier-free guidance
|
198 |
generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
|
199 |
batch_size = 1
|
@@ -279,7 +279,7 @@ class Styles_paintings():
|
|
279 |
def generate_styles_with_custom_loss(self, image):
|
280 |
height = 512 # default height of Stable Diffusion
|
281 |
width = 512 # default width of Stable Diffusion
|
282 |
-
num_inference_steps =
|
283 |
guidance_scale = 8 #@param # Scale for classifier-free guidance
|
284 |
batch_size = 1
|
285 |
custom_loss_scale = 200 #@param
|
|
|
193 |
def generate_with_embs_custom(text_embeddings,seed):
|
194 |
height = 512 # default height of Stable Diffusion
|
195 |
width = 512 # default width of Stable Diffusion
|
196 |
+
num_inference_steps = 30 # Number of denoising steps
|
197 |
guidance_scale = 7.5 # Scale for classifier-free guidance
|
198 |
generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
|
199 |
batch_size = 1
|
|
|
279 |
def generate_styles_with_custom_loss(self, image):
|
280 |
height = 512 # default height of Stable Diffusion
|
281 |
width = 512 # default width of Stable Diffusion
|
282 |
+
num_inference_steps = 50 #@param # Number of denoising steps
|
283 |
guidance_scale = 8 #@param # Scale for classifier-free guidance
|
284 |
batch_size = 1
|
285 |
custom_loss_scale = 200 #@param
|