Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
@@ -64,7 +64,7 @@ class StreamMultiDiffusion(nn.Module):
|
|
64 |
preprocess_mask_cover_alpha: float = 0.3, # TODO
|
65 |
prompt_queue_capacity: int = 256,
|
66 |
mask_type: Literal['discrete', 'semi-continuous', 'continuous'] = 'continuous',
|
67 |
-
use_xformers: bool =
|
68 |
) -> None:
|
69 |
super().__init__()
|
70 |
|
|
|
64 |
preprocess_mask_cover_alpha: float = 0.3, # TODO
|
65 |
prompt_queue_capacity: int = 256,
|
66 |
mask_type: Literal['discrete', 'semi-continuous', 'continuous'] = 'continuous',
|
67 |
+
use_xformers: bool = False,
|
68 |
) -> None:
|
69 |
super().__init__()
|
70 |
|