Lumina-Image-2.0

Paused

paratudo commited on 1 day ago

Commit

162c17c

verified ·

1 Parent(s): 95f8c36

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ from transformers import (
 default_system_prompt = "You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts."
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "Alpha-VLLM/Lumina-Image-2.0"
-transformer_repo_id = "benjamin-paine/Lumina-Image-2.0" # Temporarily fixed, change when main repo gets updated
 if torch.cuda.is_available():
     torch_dtype = torch.bfloat16
 else:

 default_system_prompt = "You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts."
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "Alpha-VLLM/Lumina-Image-2.0"
+transformer_repo_id = "Alpha-VLLM/Lumina-Image-2.0" # Temporarily fixed, change when main repo gets updated
 if torch.cuda.is_available():
     torch_dtype = torch.bfloat16
 else: