Hugging Face's logo

Spaces:
waloneai
/
flux-lab-light
Sleeping

App Files Community
vilarin commited on
Commit
5e2e591
·
verified ·
1 Parent(s): 8a23466

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ class ModelWrapper:
90
  else:
91
  raise NotImplementedError()
92
 
93
- noise = noise.to(torch.float16)
94
  print(f'noise: {noise.dtype}')
95
  prompt_embed = prompt_embed.to(torch.float32)
96
  DTYPE = prompt_embed.dtype
 
90
  else:
91
  raise NotImplementedError()
92
 
93
+ #noise = noise.to(torch.float16)
94
  print(f'noise: {noise.dtype}')
95
  prompt_embed = prompt_embed.to(torch.float32)
96
  DTYPE = prompt_embed.dtype