ford442 commited on
Commit
f51f2cf
·
verified ·
1 Parent(s): 7b4baf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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",