Hugging Face's logo

Spaces:
Noename
/
Colorization
Paused

App Files Community
Noename commited on
Commit
2e2e1d3
·
verified ·
1 Parent(s): 463ebea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
30
  pretrained_model = 'runwayml/stable-diffusion-v1-5'
31
  controlnet = 'models'
32
  # controlnet = 'checkpoint-34000/controlnet'
33
- precision = 'bf16'
34
 
35
  # Check for different hardware architectures
36
  if torch.cuda.is_available():
 
30
  pretrained_model = 'runwayml/stable-diffusion-v1-5'
31
  controlnet = 'models'
32
  # controlnet = 'checkpoint-34000/controlnet'
33
+ precision = 'fp16'
34
 
35
  # Check for different hardware architectures
36
  if torch.cuda.is_available():