ford442 commited on
Commit
5d418db
·
verified ·
1 Parent(s): f588f87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -201,9 +201,9 @@ def load_and_prepare_model():
201
  #pipe.set_adapters(["skin", "photo", "fantasy"], adapter_weights=[0.75, 0.25, 0.5])
202
  #pipe.set_adapters(["skin"], adapter_weights=[0.5])
203
  #pipe.unet.set_default_attn_processor()
204
- pipe.to(device, torch.bfloat16)
205
- #pipe.to(device)
206
- #pipe.to(torch.bfloat16)
207
 
208
  print(f'Pipeline: ')
209
  #print(f'_optional_components: {pipe._optional_components}')
 
201
  #pipe.set_adapters(["skin", "photo", "fantasy"], adapter_weights=[0.75, 0.25, 0.5])
202
  #pipe.set_adapters(["skin"], adapter_weights=[0.5])
203
  #pipe.unet.set_default_attn_processor()
204
+ #pipe.to(device, torch.bfloat16)
205
+ pipe.to(device)
206
+ pipe.to(torch.bfloat16)
207
 
208
  print(f'Pipeline: ')
209
  #print(f'_optional_components: {pipe._optional_components}')