yanze commited on
Commit
dd6fe43
β€’
1 Parent(s): 1b1a4a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -86,6 +86,8 @@ def generate_image(
86
  id_embeddings = None
87
  uncond_id_embeddings = None
88
 
 
 
89
  # prepare input
90
  x = get_noise(
91
  1,
@@ -95,6 +97,7 @@ def generate_image(
95
  dtype=torch.bfloat16,
96
  seed=opts.seed,
97
  )
 
98
  timesteps = get_schedule(
99
  opts.num_steps,
100
  x.shape[-1] * x.shape[-2] // 4,
 
86
  id_embeddings = None
87
  uncond_id_embeddings = None
88
 
89
+ print(id_embeddings)
90
+
91
  # prepare input
92
  x = get_noise(
93
  1,
 
97
  dtype=torch.bfloat16,
98
  seed=opts.seed,
99
  )
100
+ print(x)
101
  timesteps = get_schedule(
102
  opts.num_steps,
103
  x.shape[-1] * x.shape[-2] // 4,