Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
2359407
1
Parent(s):
43eb4b6
upload
Browse files
app.py
CHANGED
|
@@ -273,6 +273,7 @@ def main():
|
|
| 273 |
|
| 274 |
vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae')
|
| 275 |
unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1")
|
|
|
|
| 276 |
empty_text_embed = torch.from_numpy(np.load("./empty_text_embed.npy")).to(device, dtype)[None] # [1, 77, 1024]
|
| 277 |
|
| 278 |
pipe = GenPerceptPipeline(vae=vae,
|
|
|
|
| 273 |
|
| 274 |
vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae')
|
| 275 |
unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1")
|
| 276 |
+
dtype = unet_depth_v1.dtype
|
| 277 |
empty_text_embed = torch.from_numpy(np.load("./empty_text_embed.npy")).to(device, dtype)[None] # [1, 77, 1024]
|
| 278 |
|
| 279 |
pipe = GenPerceptPipeline(vae=vae,
|