Kurumi commited on
Commit ·
7a1f675
1
Parent(s): ff6166b
fix: cant run float16 with cpu
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from flask import (
|
|
| 6 |
send_file,
|
| 7 |
)
|
| 8 |
from io import BytesIO
|
| 9 |
-
from torch import cuda
|
| 10 |
from urllib.parse import unquote
|
| 11 |
|
| 12 |
|
|
@@ -24,7 +24,6 @@ def main():
|
|
| 24 |
|
| 25 |
pipeline = AutoPipelineForText2Image.from_pretrained(
|
| 26 |
pretrained_model_or_path="Lykon/DreamShaper",
|
| 27 |
-
torch_dtype=float16,
|
| 28 |
)
|
| 29 |
|
| 30 |
if cuda.is_available():
|
|
|
|
| 6 |
send_file,
|
| 7 |
)
|
| 8 |
from io import BytesIO
|
| 9 |
+
from torch import cuda
|
| 10 |
from urllib.parse import unquote
|
| 11 |
|
| 12 |
|
|
|
|
| 24 |
|
| 25 |
pipeline = AutoPipelineForText2Image.from_pretrained(
|
| 26 |
pretrained_model_or_path="Lykon/DreamShaper",
|
|
|
|
| 27 |
)
|
| 28 |
|
| 29 |
if cuda.is_available():
|