Hugging Face's logo

ryderbg
/
stable_diffusion

Model card Files Community
Nedyalko Demirev commited on
Commit
e2aa61c
•
1 Parent(s): 815ab0f

ezreal2211

Browse files
Files changed (2) hide show
  1. nvidia.py +1 -1
  2. pictures/ezreal2211.png +0 -0
nvidia.py CHANGED
@@ -7,7 +7,7 @@ def execute_nvidia(prompt, filename):
7
  pipe = StableDiffusionPipeline.from_pretrained(
8
  model_id,
9
  use_auth_token=True,
10
- # torch_dtype=torch.float16
11
  ).to(DEVICE)
12
  pipe.enable_attention_slicing()
13
  image = pipe(prompt).images[0]
 
7
  pipe = StableDiffusionPipeline.from_pretrained(
8
  model_id,
9
  use_auth_token=True,
10
+ torch_dtype=torch.float16
11
  ).to(DEVICE)
12
  pipe.enable_attention_slicing()
13
  image = pipe(prompt).images[0]
pictures/ezreal2211.png ADDED