Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
547b516
1
Parent(s):
2509eb1
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ from custom_pipeline import CosStableDiffusionXLInstructPix2PixPipeline
|
|
| 4 |
import numpy as np
|
| 5 |
import math
|
| 6 |
import spaces
|
|
|
|
| 7 |
|
| 8 |
edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
|
| 9 |
normal_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl.safetensors")
|
|
|
|
| 4 |
import numpy as np
|
| 5 |
import math
|
| 6 |
import spaces
|
| 7 |
+
import torch
|
| 8 |
|
| 9 |
edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
|
| 10 |
normal_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl.safetensors")
|