Spaces:

alfredplpl
/

sd3-with-LLM

Running on Zero

alfredplpl commited on Oct 22, 2024

Commit

08c5790

•

1 Parent(s): a423cc4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ device = "cuda"
 dtype = torch.float16
 repo = "stabilityai/stable-diffusion-3.5-large"
-t2i = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16, token=os.environ["TOKEN"]).to(device)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3.5-mini-instruct",

 dtype = torch.float16
 repo = "stabilityai/stable-diffusion-3.5-large"
+t2i = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.bfloat16, token=os.environ["TOKEN"]).to(device)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3.5-mini-instruct",