treasuraid commited on
Commit
0185ba1
1 Parent(s): 73ba89d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def load_pipeline_w_lora() :
22
  pipeline = DiffusionPipeline.from_pretrained(
23
  MODEL_REPO,
24
  revision=None,
25
- torch_dtype=torch.float16,
26
  )
27
 
28
  # Load LoRa attn layer weights to unet attn layers
 
22
  pipeline = DiffusionPipeline.from_pretrained(
23
  MODEL_REPO,
24
  revision=None,
25
+ torch_dtype=torch.float32,
26
  )
27
 
28
  # Load LoRa attn layer weights to unet attn layers