leemonz commited on
Commit
d2871ea
·
verified ·
1 Parent(s): 4256ae3

Upload 3 files

Browse files
Files changed (2) hide show
  1. README.md +26 -8
  2. app.py +2 -2
README.md CHANGED
@@ -1,17 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Generador con SDXL + LoRA en GPU T4 Gratis
2
 
3
  Este Space usa:
4
- - Modelo base: stabilityai/stable-diffusion-xl-base-1.0
5
- - LoRA de ejemplo: nerijs/pixel-art-xl
6
 
7
  ## Cómo cambiar el LoRA
8
- 1. Sube tu LoRA a Hugging Face (como modelo).
9
- 2. En `app.py`, reemplaza:
 
10
  LORA_MODEL = "nerijs/pixel-art-xl"
11
- por tu repo: "usuario/mi-lora"
 
 
 
 
12
  3. Guarda y vuelve a lanzar el Space.
13
 
14
  ## Recomendaciones para GPU T4
15
- - Usar `torch_dtype=torch.float16`
16
- - 20–30 pasos de inferencia
17
- - Mantener LoRAs livianos para evitar OOM
 
1
+
2
+ ---
3
+ title: Generador SDXL + LoRA (T4)
4
+ emoji: 🚀
5
+ colorFrom: indigo
6
+ colorTo: purple
7
+ sdk: gradio
8
+ sdk_version: "4.0.0"
9
+ app_file: app.py
10
+ python_version: 3.10
11
+ pinned: false
12
+ ---
13
+
14
  # Generador con SDXL + LoRA en GPU T4 Gratis
15
 
16
  Este Space usa:
17
+ - Modelo base: `stabilityai/stable-diffusion-xl-base-1.0`
18
+ - LoRA de ejemplo: `nerijs/pixel-art-xl`
19
 
20
  ## Cómo cambiar el LoRA
21
+ 1. Sube tu LoRA a Hugging Face (como **Model**).
22
+ 2. En `app.py`, reemplaza la línea:
23
+ ```python
24
  LORA_MODEL = "nerijs/pixel-art-xl"
25
+ ```
26
+ por tu repo, por ejemplo:
27
+ ```python
28
+ LORA_MODEL = "usuario/mi-lora"
29
+ ```
30
  3. Guarda y vuelve a lanzar el Space.
31
 
32
  ## Recomendaciones para GPU T4
33
+ - Mantener `torch_dtype=torch.float16`.
34
+ - 20–30 pasos de inferencia.
35
+ - Evitar LoRAs muy pesados para prevenir OOM.
app.py CHANGED
@@ -3,9 +3,9 @@ from diffusers import StableDiffusionXLPipeline
3
  import gradio as gr
4
 
5
  # Modelo base
6
- BASE_MODEL = "dhead/wai-nsfw-illustrious-sdxl-v140-sdxl"
7
  # LoRA de ejemplo (puedes cambiarlo por el tuyo)
8
- LORA_MODEL = "leemonz/ROSPRITE"
9
 
10
  print("Cargando modelo base...")
11
  pipe = StableDiffusionXLPipeline.from_pretrained(
 
3
  import gradio as gr
4
 
5
  # Modelo base
6
+ BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
7
  # LoRA de ejemplo (puedes cambiarlo por el tuyo)
8
+ LORA_MODEL = "nerijs/pixel-art-xl"
9
 
10
  print("Cargando modelo base...")
11
  pipe = StableDiffusionXLPipeline.from_pretrained(