Update app.py
Browse files
app.py
CHANGED
|
@@ -11,8 +11,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 11 |
pipe = DiffusionPipeline.from_pretrained(
|
| 12 |
"CompVis/stable-diffusion-v1-4",
|
| 13 |
use_auth_token=MY_SECRET_TOKEN,
|
| 14 |
-
revision='fp16',
|
| 15 |
-
torch_dtype=torch.float16,
|
| 16 |
safety_checker=None, # Very important for videos...lots of false positives while interpolating
|
| 17 |
custom_pipeline="interpolate_stable_diffusion",
|
| 18 |
).to(device)
|
|
|
|
| 11 |
pipe = DiffusionPipeline.from_pretrained(
|
| 12 |
"CompVis/stable-diffusion-v1-4",
|
| 13 |
use_auth_token=MY_SECRET_TOKEN,
|
| 14 |
+
#revision='fp16',
|
| 15 |
+
#torch_dtype=torch.float16,
|
| 16 |
safety_checker=None, # Very important for videos...lots of false positives while interpolating
|
| 17 |
custom_pipeline="interpolate_stable_diffusion",
|
| 18 |
).to(device)
|