Hugging Face's logo

Spaces:
Kitasurfer
/
stable-diffusion-webui
Runtime error

App Files Community
Kitasurfer commited on
Commit
e7e4f84
·
verified ·
1 Parent(s): dba16f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
 
5
- # Загрузка модели с Hugging Face
6
- model_id = "lllyasviel/flux1_dev"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
 
9
  def generate_image(prompt):
 
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
 
5
+ # Используем модель lllyasviel/flux1-dev-bnb-nf4
6
+ model_id = "lllyasviel/flux1-dev-bnb-nf4"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
 
9
  def generate_image(prompt):