Ahsen Khaliq commited on
Commit
45c31f1
1 Parent(s): ed2f52e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, image_prompt,timestep_respacing):
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 = [image_prompt.name]
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