Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
45c31f1
1
Parent(s):
ed2f52e
Update app.py
Browse files
app.py
CHANGED
|
@@ -78,7 +78,7 @@ def tv_loss(input):
|
|
| 78 |
def range_loss(input):
|
| 79 |
return (input - input.clamp(-1, 1)).pow(2).mean([1, 2, 3])
|
| 80 |
|
| 81 |
-
def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed,
|
| 82 |
# Model settings
|
| 83 |
model_config = model_and_diffusion_defaults()
|
| 84 |
model_config.update({
|
|
@@ -119,7 +119,7 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
|
|
| 119 |
all_frames = []
|
| 120 |
prompts = [text]
|
| 121 |
if image_prompts:
|
| 122 |
-
image_prompts = [
|
| 123 |
else:
|
| 124 |
image_prompts = []
|
| 125 |
batch_size = 1
|
|
|
|
| 78 |
def range_loss(input):
|
| 79 |
return (input - input.clamp(-1, 1)).pow(2).mean([1, 2, 3])
|
| 80 |
|
| 81 |
+
def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed, image_prompts,timestep_respacing):
|
| 82 |
# Model settings
|
| 83 |
model_config = model_and_diffusion_defaults()
|
| 84 |
model_config.update({
|
|
|
|
| 119 |
all_frames = []
|
| 120 |
prompts = [text]
|
| 121 |
if image_prompts:
|
| 122 |
+
image_prompts = [image_prompts.name]
|
| 123 |
else:
|
| 124 |
image_prompts = []
|
| 125 |
batch_size = 1
|