huggingface / huggingface/diffusers

[Community] Only half of my cpu cores are being used?

Ouverte
#1,109 2 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
bug Good second issue
Langage dominant
Python
Étoiles
34.5k
Forks
7.3k
Merge moyen
3 j 3 h
PR mergées (30 j)
91

Description

### Describe the bug

Only about half of my cpus are used.
I have 8 cores, but only 4 are used. Is there way to fix this?
![image](https://user-images.githubusercontent.com/55401336/199564591-b57f9212-6346-4062-b2cc-a11c67fea030.png)
![image](https://user-images.githubusercontent.com/55401336/199564675-cee7a67b-53cc-4e89-a690-1f393d2ed95d.png)
^ The spikes are when diffussers is generating something

### Reproduction

My code:
```python
print("Hello, World")

import secrets
import gradio as gr
import torch
from torch import autocast
from diffusers import StableDiffusionPipeline
from PIL import Image

print("Deps loaded!")

model_id = "CompVis/stable-diffusion-v1-4"
device = "cpu"

pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
pipe = pipe.to(device)

print("Loaded!")

def predict(name):
print(f"Prompt: {name}")
prompt = name
with autocast("cuda"):
image = pipe(prompt, guidance_scale=7.5, width=512, height=512, num_inference_steps=20).images[0]

id = secrets.token_urlsafe(16)
image.save(f"./out/{id}.png")

return image

print("Starting...")
demo = gr.Interface(
predict,
inputs=[
gr.inputs.Textbox(label='Prompt', default='a chalk pastel drawing of a llama wearing a wizard hat')
],
outputs=gr.Image(shape=[512,512], type="pil", elem_id="output_image"),
css="#output_image{width: 512px; height: 512px}",
title="Retslav - Text To Image - Stable Diffusion",
description="Retslav Stable Diffussion",
)

demo.launch(server_port=3000)
```

### Logs

```shell
-
```

### System Info

- `diffusers` version: 0.6.0
- Platform: Linux-5.4.0-125-generic-x86_64-with-glibc2.29
- Python version: 3.8.10
- PyTorch version (GPU?): 1.13.0+cpu (False)
- Huggingface_hub version: 0.10.1
- Transformers version: 4.23.1
- Using GPU in script?: NO
- Using distributed or parallel set-up in script?: NO

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Commencez par le script de reproduction fourni, en particulier StableDiffusionPipeline et le chemin .to("cpu"), et reproduisez l’utilisation signalée avec les versions indiquées de Python, PyTorch, diffusers et Linux. Examinez la configuration de la génération sur CPU et de l’utilisation des threads, puis documentez une cause confirmée ainsi qu’une correction ou une limitation reproductible ; le travail est considéré comme terminé lorsque la génération utilise les cœurs disponibles attendus ou que le comportement est clairement expliqué.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, pytorch
Domaine
machine-learning, performance
Type d'issue
Bug
Difficulté
4/5
Temps estimé
3-5 jours
Activité
Calme
Clarté
À clarifier
Accessibilité débutants
35/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.