Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,6 @@ torch.backends.cudnn.benchmark = False
|
|
34 |
|
35 |
hftoken = os.getenv("HF_AUTH_TOKEN")
|
36 |
|
37 |
-
|
38 |
code = r'''
|
39 |
import torch
|
40 |
import paramiko
|
@@ -56,13 +55,11 @@ def upload_to_ftp(filename):
|
|
56 |
print(f"Uploaded {filename} to FTP server")
|
57 |
except Exception as e:
|
58 |
print(f"FTP upload error: {e}")
|
59 |
-
|
60 |
'''
|
61 |
|
62 |
pyx = cyper.inline(code, fast_indexing=True, directives=dict(boundscheck=False, wraparound=False, language_level=3))
|
63 |
|
64 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
65 |
-
torch_dtype = torch.bfloat16
|
66 |
|
67 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
68 |
#"stabilityai # stable-diffusion-3.5-large",
|
|
|
34 |
|
35 |
hftoken = os.getenv("HF_AUTH_TOKEN")
|
36 |
|
|
|
37 |
code = r'''
|
38 |
import torch
|
39 |
import paramiko
|
|
|
55 |
print(f"Uploaded {filename} to FTP server")
|
56 |
except Exception as e:
|
57 |
print(f"FTP upload error: {e}")
|
|
|
58 |
'''
|
59 |
|
60 |
pyx = cyper.inline(code, fast_indexing=True, directives=dict(boundscheck=False, wraparound=False, language_level=3))
|
61 |
|
62 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
63 |
|
64 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
65 |
#"stabilityai # stable-diffusion-3.5-large",
|