Update app.py
Browse files
app.py
CHANGED
|
@@ -126,7 +126,6 @@ def get_pipeline():
|
|
| 126 |
scheduler=EulerAncestralDiscreteScheduler.from_pretrained(MODEL_ID, subfolder="scheduler"),
|
| 127 |
torch_dtype=DTYPE,
|
| 128 |
low_cpu_mem_usage=True,
|
| 129 |
-
variant="fp16",
|
| 130 |
)
|
| 131 |
|
| 132 |
# CPU 优化
|
|
|
|
| 126 |
scheduler=EulerAncestralDiscreteScheduler.from_pretrained(MODEL_ID, subfolder="scheduler"),
|
| 127 |
torch_dtype=DTYPE,
|
| 128 |
low_cpu_mem_usage=True,
|
|
|
|
| 129 |
)
|
| 130 |
|
| 131 |
# CPU 优化
|