Alibrown commited on
Commit
8f7b856
·
verified ·
1 Parent(s): f7c1363

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -28,7 +28,8 @@ TORCH_DTYPE = os.getenv('TORCH_DTYPE', 'float16') # Standard-Präzision
28
 
29
  # Modell laden
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
- torch_dtype = torch.float16 if TORCH_DTYPE == 'float16' else torch.float32
 
32
 
33
  try:
34
  pipe = StableDiffusionPipeline.from_pretrained(
 
28
 
29
  # Modell laden
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
+ torch_dtype = torch.float16 if torch.cuda.is_available() and TORCH_DTYPE == 'float16' else torch.float32
32
+
33
 
34
  try:
35
  pipe = StableDiffusionPipeline.from_pretrained(