Update app.py

app.py

@@ -11,7 +11,7 @@ from PIL import Image
 
 
 MODEL_REPO = 'OFA-Sys/small-stable-diffusion-v0'
-LoRa_DIR = 'VESSL/lora-fintuning-bayc'
+LoRa_DIR = 'weights'
 DATASET_REPO = 'VESSL/Bored_Ape_NFT_text'
 SAMPLE_IMAGE = 'weights/Sample.png'
 
@@ -22,7 +22,7 @@ def load_pipeline_w_lora() :
     pipeline = DiffusionPipeline.from_pretrained(
         MODEL_REPO,
         revision=None,
-        torch_dtype=torch.float32,
+        torch_dtype=torch.float16,
     )
     
     # Load LoRa attn layer weights to unet attn layers