Hugging Face's logo

Spaces:
ironjr
/
SemanticPalette
Runtime error

App Files Community
1
ironjr commited on
Commit
32502c8
·
verified ·
1 Parent(s): 932fb9b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -38,7 +38,7 @@ class StableMultiDiffusionPipeline(nn.Module):
38
  def __init__(
39
  self,
40
  device: torch.device,
41
- dtype: torch.dtype = torch.float32,
42
  sd_version: Literal['1.5', '2.0', '2.1', 'xl'] = '1.5',
43
  hf_key: Optional[str] = None,
44
  lora_key: Optional[str] = None,
 
38
  def __init__(
39
  self,
40
  device: torch.device,
41
+ dtype: torch.dtype = torch.float16,
42
  sd_version: Literal['1.5', '2.0', '2.1', 'xl'] = '1.5',
43
  hf_key: Optional[str] = None,
44
  lora_key: Optional[str] = None,