Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
13 |
|
14 |
# # notebook_login(token_key)
|
15 |
|
16 |
-
|
17 |
if torch.cuda.is_available():
|
18 |
torch.cuda.max_memory_allocated(device=device)
|
19 |
pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
|
|
13 |
|
14 |
# # notebook_login(token_key)
|
15 |
|
16 |
+
@spaces.GPU
|
17 |
if torch.cuda.is_available():
|
18 |
torch.cuda.max_memory_allocated(device=device)
|
19 |
pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|