GrayShine commited on
Commit
112d595
β€’
1 Parent(s): 0723018

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -181,8 +181,6 @@ def init_model():
181
  model = get_models(args).to(device)
182
  model = tca_transform_model(model).to(device)
183
  model = ip_transform_model(model).to(device)
184
- if args.use_compile:
185
- model = torch.compile(model)
186
  if args.enable_xformers_memory_efficient_attention:
187
  if is_xformers_available():
188
  model.enable_xformers_memory_efficient_attention()
 
181
  model = get_models(args).to(device)
182
  model = tca_transform_model(model).to(device)
183
  model = ip_transform_model(model).to(device)
 
 
184
  if args.enable_xformers_memory_efficient_attention:
185
  if is_xformers_available():
186
  model.enable_xformers_memory_efficient_attention()