nalin0503 commited on
Commit
6de4b7d
·
1 Parent(s): 9bb1740
Files changed (1) hide show
  1. Image-Morpher/main.py +1 -1
Image-Morpher/main.py CHANGED
@@ -124,7 +124,7 @@ pipeline.enable_xformers_memory_efficient_attention()
124
 
125
  from accelerate import init_empty_weights
126
  with init_empty_weights():
127
- pipeline.to("cuda", dtype=torch.float32)
128
 
129
  # Add these AFTER device movement
130
  torch.backends.cudnn.benchmark = True # finds efficient convolution algo by running short benchmark, minimal speed-up.
 
124
 
125
  from accelerate import init_empty_weights
126
  with init_empty_weights():
127
+ pipeline.to("cuda")
128
 
129
  # Add these AFTER device movement
130
  torch.backends.cudnn.benchmark = True # finds efficient convolution algo by running short benchmark, minimal speed-up.