Hugging Face's logo

Spaces:
Thuchk
/
Stable_diffusion_v1.4
Runtime error

App Files Community
Thuchk commited on
Commit
17933fc
•
1 Parent(s): 1d2e56a

fix low memory issue

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py → gradio_app.py +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🤗
4
  colorFrom: yellow
5
  colorTo: orange
6
  sdk: gradio
7
- app_file: app.py
8
  pinned: false
9
  ---
10
 
 
4
  colorFrom: yellow
5
  colorTo: orange
6
  sdk: gradio
7
+ app_file: gradio_app.py
8
  pinned: false
9
  ---
10
 
app.py → gradio_app.py RENAMED
@@ -5,7 +5,7 @@ from PIL import Image
5
  import gradio as gr
6
 
7
  model_path = "sd-pokemon-model"
8
- pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
9
  pipe.to("cuda")
10
 
11
  def predict(prompt):
 
5
  import gradio as gr
6
 
7
  model_path = "sd-pokemon-model"
8
+ pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16, low_cpu_mem_usage=False, device_map=None)
9
  pipe.to("cuda")
10
 
11
  def predict(prompt):