Hugging Face's logo

Spaces:
rphrp1985
/
stable-diffusion-3-medium
Running on Zero

App Files Community
rphrp1985 commited on
Commit
adc2760
•
1 Parent(s): 1be7823

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ dtype = torch.float16
12
 
13
  repo = "dataautogpt3/OpenDalleV1.1"
14
  repo = "stabilityai/stable-diffusion-3-medium"
 
15
 
16
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
17
  # pipeline = AutoPipelineForText2Image.from_pretrained(repo, torch_dtype=torch.float16).to('cuda')
 
12
 
13
  repo = "dataautogpt3/OpenDalleV1.1"
14
  repo = "stabilityai/stable-diffusion-3-medium"
15
+ repo="stabilityai/stable-diffusion-3-medium-tensorrt"
16
 
17
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
18
  # pipeline = AutoPipelineForText2Image.from_pretrained(repo, torch_dtype=torch.float16).to('cuda')