Spaces:
Configuration error
Configuration error
Update app.py
Browse files
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()
|