Spaces:

alfredplpl
/

sd3-with-LLM

Sleeping

alfredplpl commited on Jun 12

Commit

0a53e22

•

1 Parent(s): c5d561f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ dtype = torch.float16
 repo = "stabilityai/stable-diffusion-3-medium"
 t2i = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="refs/pr/26",token=os.environ["TOKEN"]).to(device)
-model_id = "microsoft/Phi-3-small-128k-instruct"
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map=device,

 repo = "stabilityai/stable-diffusion-3-medium"
 t2i = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="refs/pr/26",token=os.environ["TOKEN"]).to(device)
+model_id = "microsoft/Phi-3-medium-128k-instruct"
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map=device,