Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
5a557d2
·
verified ·
1 Parent(s): 3968f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -79,7 +79,8 @@ class ModelWrapper:
79
 
80
  @spaces.GPU()
81
  def sample(self, noise, unet_added_conditions, prompt_embed, fast_vae_decode):
82
- alphas_cumprod = self.scheduler.alphas_cumprod.to(self.device)
 
83
  if self.num_step == 1:
84
  all_timesteps = [self.conditioning_timestep]
85
  step_interval = 0
 
79
 
80
  @spaces.GPU()
81
  def sample(self, noise, unet_added_conditions, prompt_embed, fast_vae_decode):
82
+ #alphas_cumprod = self.scheduler.alphas_cumprod.to(self.device)
83
+ print("sampling...")
84
  if self.num_step == 1:
85
  all_timesteps = [self.conditioning_timestep]
86
  step_interval = 0