ironjr commited on
Commit
d057fb7
1 Parent(s): 5b53c09

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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 = True,
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