Spaces:

alfredplpl
/

sd3-with-LLM

Running on Zero

alfredplpl commited on Jun 12, 2024

Commit

0ef6574

verified ·

1 Parent(s): 718bcd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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