Spaces:
Runtime error
Runtime 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()
|