Spaces:
Runtime error
Runtime error
chandrasekhar319
commited on
Commit
•
bcef729
1
Parent(s):
7827ab5
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ if torch.cuda.is_available():
|
|
12 |
pipe.enable_xformers_memory_efficient_attention()
|
13 |
pipe = pipe.to(device)
|
14 |
else:
|
15 |
-
pipe = DiffusionPipeline.from_pretrained("/home/user/app/
|
16 |
pipe = pipe.to(device)
|
17 |
|
18 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
12 |
pipe.enable_xformers_memory_efficient_attention()
|
13 |
pipe = pipe.to(device)
|
14 |
else:
|
15 |
+
pipe = DiffusionPipeline.from_pretrained("/home/user/app/stable_diffusion_pipeline.pth", use_safetensors=True)
|
16 |
pipe = pipe.to(device)
|
17 |
|
18 |
MAX_SEED = np.iinfo(np.int32).max
|