rphrp1985 commited on
Commit
77197c9
·
verified ·
1 Parent(s): b70eb21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
11
  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')
 
11
  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')