Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ from diffusers import DiffusionPipeline, UNet2DConditionModel, FlowMatchEulerDis
|
|
| 7 |
from typing import Optional, Union, List, Tuple
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
-
|
| 11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 12 |
dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 13 |
model_repo_id = "AiArtLab/sdxs-1b"
|
|
|
|
| 7 |
from typing import Optional, Union, List, Tuple
|
| 8 |
from PIL import Image
|
| 9 |
|
|
|
|
| 10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 11 |
dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 12 |
model_repo_id = "AiArtLab/sdxs-1b"
|