cuda in device

app.py

@@ -4,14 +4,14 @@ import random
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import AutoPipelineForText2Image
 import torch
+from huggingface_hub import snapshot_download
 
 
-from huggingface_hub import snapshot_download
-snapshot_download(repo_id="roomie/xavyy", cache_dir='./')
+snapshot_download(repo_id="Roomie/xavyy", cache_dir='./')
 
 pipeline = AutoPipelineForText2Image.from_pretrained(
     'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cpu')
-pipeline.load_lora_weights('roomie/xavyy', weight_name='xavyy.safetensors')
+pipeline.load_lora_weights('Roomie/xavyy', weight_name='xavyy.safetensors')
 
 
 MAX_SEED = np.iinfo(np.int32).max

requirements.txt

@@ -4,4 +4,5 @@ invisible_watermark
 torch
 transformers
 xformers
-sentencepiece
+sentencepiece
+peft